Update app.py

app.py

@@ -23,7 +23,7 @@ def ask_chatbot(query, chat_history, project_data):
     # Convert project_data to a string format (optional: simplify or summarize if needed)
     project_data_str = json.dumps(project_data, indent=2)
 
-    if len(chat_history) >= 5:
+    if len(chat_history) >= 10:
         chat_history.pop(0)
 
     query_with_history = ""
@@ -65,15 +65,25 @@ def generate_response(message, history):
 def upload_file(data_file):
   with open(data_file.name, "r") as f:
     data = json.load(f)
-  gr.Info("Prooject file Uploaded. You can now query the document")
+  gr.Info("Project file Uploaded. You can now query the document")
   with open("/content/project_data.json", "w") as f:
     json.dump(data, f)
 
 with gr.Blocks() as demo:
-  with gr.Column():
+    with gr.Column():
     u = gr.UploadButton("Upload a file", file_count="single")
     u.upload(upload_file, u)
-  gr.ChatInterface(generate_response)
+    gr.ChatInterface(
+        yes_man,
+        chatbot=gr.Chatbot(height=300),
+        title="AI Project Manager",
+        description="Virtual Project Management Assistant",
+        theme="soft",
+        examples=["Which task has the earliest deadline?", "What is the current progress?"],
+        cache_examples=True,
+        undo_btn="Delete Previous",
+        clear_btn="Clear",
+    )
 
 demo.launch()