Upload folder using huggingface_hub

agent/research_agent.py

@@ -102,7 +102,7 @@ class ResearchAgent:
         Args: None
         Returns: str: The report for the given question
         """
-        yield "Searching online..."
+        # yield "Searching online..."
 
         report_type_func = prompts.get_report_by_type(report_type)
         

app.py

@@ -5,17 +5,21 @@ from agent.research_agent import ResearchAgent
 from agent.toolkits import english_polishing
 from statics.style import *
 
-theme = gr.themes.Soft(
-    primary_hue=gr.themes.Color(c100="#e0e7ff", c200="#c7d2fe", c300="#a5b4fc", c400="#818cf8", c50="#eef2ff", c500="#6366f1", c600="#5e5aaa", c700="#4338ca", c800="#3730a3", c900="#312e81", c950="#2b2c5e"),
-    font_mono=[gr.themes.GoogleFont('Fira Code'), 'ui-monospace', 'Consolas', 'monospace']
-)
 
 check_openai_api_key()
+history = ""
+history_num = 0
+history_tasks = []
+max_histroy_num = 30
 
 def run_agent(task, agent, report_type):
+    global history, history_num, history_tasks
+    history_num += 1
+    history_tasks.append(task)
     assistant = ResearchAgent(task, agent)
     yield from assistant.write_report(report_type)
 
+
 with gr.Blocks(theme=gr.themes.Base(),
                title="AI Research Assistant",
                css=css) as demo:
@@ -44,17 +48,33 @@ with gr.Blocks(theme=gr.themes.Base(),
                                          interactive=True,
                                          allow_custom_value=False,
                                          choices=["Research Report",
-                                                 "Resource Report",
-                                                 "Outline Report"])
+                                                  "Resource Report",
+                                                  "Outline Report"])
+            
             input_box = gr.Textbox(label="# What would you like to research next?", placeholder="Enter your question here")
-            submit_btn = gr.Button("Generate Report")
-            submit_btn.click(run_agent, inputs=[input_box, agent_type, report_type], 
-                                        outputs=research_report)
+            submit_btn = gr.Button("Generate Report", elem_id="primary-btn")
+
             gr.Examples(["Should I invest in the Large Language Model industry in 2023?", 
                          "Is it advisable to make investments in the electric car industry during the year 2023?",
                          "What constitutes the optimal approach for investing in the Bitcoin industry during the year 2023?",
                          "What are the most recent advancements in the domain of superconductors as of 2023?"], 
                          inputs=input_box)
+            
+            with gr.Accordion(label="# Research History (Max 30)", elem_id="history", open=False):
+                research_history = gr.Markdown()
+            
+            def store_research(content):
+                global history_num, history_tasks, history
+                if 0 < history_num <= max_histroy_num:
+                    history += f'<details>\
+                                    <summary>Research History {history_num}: <i>{history_tasks[-1]}</i></summary>\
+                                    <div id="history_box">{content}</div>\
+                                 </details>'
+                return history
+                    
+            submit_btn.click(run_agent, inputs=[input_box, agent_type, report_type], outputs=research_report)\
+                      .then(store_research, inputs=[research_report], outputs=research_history)
+
     
     with gr.Tab("English Polishing"):
         gr.HTML(english_polishing_html)
@@ -62,8 +82,8 @@ with gr.Blocks(theme=gr.themes.Base(),
         sentences = gr.Textbox(label="# What would you like to polish?", placeholder="Enter your sentence here")
         
         with gr.Row():
-            polish_btn = gr.Button("Polish")
-            save_btn = gr.Button("Save")
+            polish_btn = gr.Button("Polish", elem_id="primary-btn")
+            save_btn = gr.Button("Save", elem_id="primary-btn")
         
         polish_btn.click(english_polishing, inputs=[sentences], outputs=polished_result)
         

statics/style.py

@@ -34,10 +34,31 @@ css = """
     .output {
         padding: 10px;
         min-height: 200px;
-        border: #c0c0c0 1px solid;
-        border-radius: 5px;
+        border: 1.5px solid #9A73B5;
+        border-radius: 10px;
         margin-bottom: 10px;
     }
+    
+    #history {
+        padding: 10px;
+        border: 1.5px dashed #9A73B5;
+        border-radius: 10px;
+    }
+    
+    #primary-btn {
+        border: 1.5px solid #9A73B5;
+        font-size: 20px;
+    }
+    
+    summary {
+        font-size: 14px;
+        font-weight: bold;
+    }
+    
+    #history_box {
+        border-bottom: 1.5px dashed #9A73B5;
+        padding: 10px;
+    }
 """
 
 top_bar = """