alonsosilva commited on
Commit
cc698c5
·
1 Parent(s): 104a23e

Order the code

Browse files
Files changed (1) hide show
  1. app.py +15 -4
app.py CHANGED
@@ -122,13 +122,23 @@ def ChatInterface():
122
  if obj['nodes'][0]['label'] not in [None, '']:
123
  for i, node in enumerate(obj['nodes']):
124
  if obj['nodes'][i]['label'] not in [None, '']:
125
- dot.node(name=str(obj['nodes'][i]['id']), label=obj['nodes'][i]['label'], color=obj['nodes'][i]['color'])
 
 
 
 
126
  if obj['edges'] not in [None, []]:
127
  if obj['edges'][0]['label'] not in [None, '']:
128
  for i, edge in enumerate(obj['edges']):
129
  if obj['edges'][i]['source'] not in [None,''] and obj['edges'][i]['target'] not in [None,''] and obj['edges'][i]['label'] not in [None,'']:
130
- dot.edge(str(obj['edges'][i]['source']), str(obj['edges'][i]['target']), label=obj['edges'][i]['label'], color=obj['edges'][i]['color'])
131
- solara.display(dot)
 
 
 
 
 
 
132
 
133
  messages: solara.Reactive[List[MessageDict]] = solara.reactive([])
134
  aux = solara.reactive("")
@@ -156,7 +166,8 @@ def Page():
156
 
157
  user_message_count = len([m for m in messages.value if m["role"] == "user"])
158
  def send():
159
- messages.value = [*messages.value, {"role": "user", "content": "Hello"}]
 
160
  def response(message):
161
  for extraction in extraction_stream:
162
  obj = extraction.model_dump()
 
122
  if obj['nodes'][0]['label'] not in [None, '']:
123
  for i, node in enumerate(obj['nodes']):
124
  if obj['nodes'][i]['label'] not in [None, '']:
125
+ dot.node(
126
+ name=str(obj['nodes'][i]['id']),
127
+ label=obj['nodes'][i]['label'],
128
+ color=obj['nodes'][i]['color']
129
+ )
130
  if obj['edges'] not in [None, []]:
131
  if obj['edges'][0]['label'] not in [None, '']:
132
  for i, edge in enumerate(obj['edges']):
133
  if obj['edges'][i]['source'] not in [None,''] and obj['edges'][i]['target'] not in [None,''] and obj['edges'][i]['label'] not in [None,'']:
134
+ dot.edge(
135
+ tail_name=str(obj['edges'][i]['source']),
136
+ head_name=str(obj['edges'][i]['target']),
137
+ label=obj['edges'][i]['label'],
138
+ color=obj['edges'][i]['color']
139
+ )
140
+ with solara.Card():
141
+ solara.display(dot)
142
 
143
  messages: solara.Reactive[List[MessageDict]] = solara.reactive([])
144
  aux = solara.reactive("")
 
166
 
167
  user_message_count = len([m for m in messages.value if m["role"] == "user"])
168
  def send():
169
+ messages.value = [{"role": "user", "content": ""}]
170
+ #messages.value = [*messages.value, {"role": "user", "content": ""}]
171
  def response(message):
172
  for extraction in extraction_stream:
173
  obj = extraction.model_dump()