yetessam commited on
Commit
deeedca
·
verified ·
1 Parent(s): 0246cbf

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +16 -0
Gradio_UI.py CHANGED
@@ -47,6 +47,20 @@ agent_footer = """
47
  Thanks for trying it out!
48
  """
49
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
50
  def pull_messages_from_step(
51
  step_log: MemoryStep,
52
  ):
@@ -197,6 +211,7 @@ def stream_to_gradio(
197
  yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
198
 
199
 
 
200
  class GradioUI:
201
  """A one-line interface to launch your agent in Gradio"""
202
 
@@ -211,6 +226,7 @@ class GradioUI:
211
  if not os.path.exists(file_upload_folder):
212
  os.mkdir(file_upload_folder)
213
 
 
214
  def interact_with_agent(self, prompt, messages):
215
  import gradio as gr
216
 
 
47
  Thanks for trying it out!
48
  """
49
 
50
+ def get_examples():
51
+ example_root = os.path.join(os.path.dirname(__file__), "examples")
52
+ # Get list of all example text paths
53
+ example_files = [os.path.join(example_root, _) for _ in os.listdir(example_root) if _.endswith("txt")]
54
+
55
+
56
+ # Read the content of each file (assuming they're text-based PDFs or plain text files)
57
+ examples = []
58
+ for file_path in example_files:
59
+ with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
60
+ examples.append(f.read()) # Read the content and append it to the list
61
+
62
+ return examples
63
+
64
  def pull_messages_from_step(
65
  step_log: MemoryStep,
66
  ):
 
211
  yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
212
 
213
 
214
+
215
  class GradioUI:
216
  """A one-line interface to launch your agent in Gradio"""
217
 
 
226
  if not os.path.exists(file_upload_folder):
227
  os.mkdir(file_upload_folder)
228
 
229
+
230
  def interact_with_agent(self, prompt, messages):
231
  import gradio as gr
232