srossitto79 commited on
Commit
d13f3dd
·
1 Parent(s): 0afa5a9
Files changed (2) hide show
  1. RBotReloaded.py +1 -1
  2. agent_llama_ui.py +3 -2
RBotReloaded.py CHANGED
@@ -338,7 +338,7 @@ AI:
338
 
339
  print(f"Action Output: {res}")
340
  observations.append(f"Action Output: {res}")
341
- prompt = prompt + f"Action: {tool.name}({action_input})\nSYS:{res}\nAI:"
342
  final_response = res # just in case it reaches max iterations
343
  else:
344
  final_response = "\n*Reasoning: ".join(observations) + f"\n{output}" if len(observations) > 0 else f"\n{output}"
 
338
 
339
  print(f"Action Output: {res}")
340
  observations.append(f"Action Output: {res}")
341
+ prompt = prompt + f"Action: {tool.name}({action_input})\nSYS: {res}\nAI:"
342
  final_response = res # just in case it reaches max iterations
343
  else:
344
  final_response = "\n*Reasoning: ".join(observations) + f"\n{output}" if len(observations) > 0 else f"\n{output}"
agent_llama_ui.py CHANGED
@@ -198,7 +198,7 @@ def render_simple_chat():
198
  unset_image_gen_guide()
199
  st.experimental_rerun()
200
  else:
201
- image_gen_guide = st.sidebar.file_uploader("Drag and Drop an image for the image generation", type=["jpg", "png"])
202
  if image_gen_guide:
203
  set_image_gen_guide(image_gen_guide)
204
  st.sidebar.success(f"File '{image_gen_guide.name}' set as image generation guidance.")
@@ -212,8 +212,9 @@ def render_simple_chat():
212
  generated_files = get_generated_files()
213
  st.sidebar.subheader("Generated Files")
214
  for file_path in generated_files:
 
215
  st.write("---")
216
- st.write(file_path.split("/")[-1].split("\\")[-1])
217
  st.image(file_path)
218
 
219
  i = 0
 
198
  unset_image_gen_guide()
199
  st.experimental_rerun()
200
  else:
201
+ image_gen_guide = st.sidebar.file_uploader("Drag and Drop an optional image to use as Guidance", type=["jpg", "png"])
202
  if image_gen_guide:
203
  set_image_gen_guide(image_gen_guide)
204
  st.sidebar.success(f"File '{image_gen_guide.name}' set as image generation guidance.")
 
212
  generated_files = get_generated_files()
213
  st.sidebar.subheader("Generated Files")
214
  for file_path in generated_files:
215
+ file_name = file_path.split("/")[-1].split("\\")[-1]
216
  st.write("---")
217
+ st.markdown(f"[{file_name}]({file_path})", unsafe_allow_html=True)
218
  st.image(file_path)
219
 
220
  i = 0