Update app.py

app.py

@@ -24,7 +24,16 @@ from prompt_template import (
 )
 
 # API Keys and Constants
-DEEPINFRA_API_KEY = "285LUJulGIprqT6hcPhiXtcrphU04FG4"
+GEMINI_API_KEY="AIzaSyCUCivstFpC9pq_jMHMYdlPrmh9Bx97dFo"
+TAVILY_API_KEY="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"
+OPENAI_API_KEY="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"
+QDRANT_API_KEY="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIiwiZXhwIjoxNzUxMDUxNzg4fQ.I9J-K7OM0BtcNKgj2d4uVM8QYAHYfFCVAyP4rlZkK2E"
+QDRANT_URL="https://6a3aade6-e8ad-4a6c-a579-21f5af90b7e8.us-east4-0.gcp.cloud.qdrant.io"
+OPENAI_API_KEY="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"
+WEAVIATE_URL="https://xbvlj5rpqyiswspww0tthq.c0.us-west3.gcp.weaviate.cloud"
+WEAVIATE_API_KEY="RU9acU1CYnNRTjY1S1ZFc18zNS9tQktaWlcwTzFEUjlscEVCUGF4YU5xRWx2MDhmTUtIdUhnOWdOTGVZPV92MjAw"
+DEEPINFRA_API_KEY="285LUJulGIprqT6hcPhiXtcrphU04FG4"
+DEEPINFRA_BASE_URL="https://api.deepinfra.com/v1/openai"
 
 # Initialize OpenAI client
 openai = OpenAI(
@@ -32,7 +41,7 @@ openai = OpenAI(
     base_url="https://api.deepinfra.com/v1/openai",
 )
 
-SESSION_ID = datetime.now().strftime("%Y%m%d_%H%M%S")
+SESSION_ID = "default"
 
 # Chat Completion Helper
 def call_llm(model: str, messages: list[dict], temperature: float = 0.0, **kwargs) -> str:
@@ -203,41 +212,47 @@ def main_pipeline_with_doc(query, doc_file, doc_type):
         return f"[Old Document Uploaded]\n{status}\n\n{answer}"
     else:
         return "Invalid document type."
+def pipeline_with_history(message, doc_file, doc_type, history):
+    response = main_pipeline_with_doc(message, doc_file, doc_type)
+    history = history + [[message, response]]
+    return history, ""
 
-with gr.Blocks(title="Wisal Main Pipeline & RAG") as demo:
-    gr.Markdown("## 🤖 Wisal: Autism AI Assistant")
-
-    with gr.Tab("🧠 Main Chatbot"):
-        with gr.Column():
-            history = gr.Textbox(label="🗨️ Chat History", lines=20, interactive=False, show_label=False)
-            
-            with gr.Row():
-                q = gr.Textbox(placeholder="Type your question here...", label="", lines=1)
-                btn = gr.Button("Send", variant="primary")
-
-            with gr.Row():
-                doc_file = gr.File(label="📎 Upload Document (PDF, DOCX, TXT)", file_types=[".pdf", ".docx", ".txt"])
-                doc_type = gr.Radio(
-                    ["None", "Knowledge Document", "User-Specific Document", "Old Document"],
-                    value="None",
-                    label="Document Type"
-                )
-
-            btn.click(
-                fn=main_pipeline_with_doc_and_history,
-                inputs=[q, doc_file, doc_type, history],
-                outputs=[history]
-            )
-
-    with gr.Tab("📘 Domain Knowledge RAG"):
-        RAG_Domain_know_doc.demo.render()
-    with gr.Tab("📁 User-Specific Documents"):
-        User_Specific_Documents.demo.render()
-    with gr.Tab("🕰️ Old Documents"):
-        Old_Document.demo.render()
+with gr.Blocks(title="Wisal Chatbot", theme=gr.themes.Base()) as demo:
+    gr.Markdown("# 🤖 Wisal: Autism AI Assistant")
+
+    chatbot = gr.Chatbot(label="Wisal Chat", height=500)
+
+    with gr.Row():
+        user_input = gr.Textbox(placeholder="Type your question here...", label="", lines=1)
+        send_btn = gr.Button("Send")
+
+    doc_file = gr.File(label="📎 Upload Document (PDF, DOCX, TXT)", file_types=[".pdf", ".docx", ".txt"])
+    doc_type = gr.Radio(
+        ["None", "Knowledge Document", "User-Specific Document", "Old Document"],
+        value="None",
+        label="Document Type"
+    )
+
+    send_btn.click(
+        fn=pipeline_with_history,
+        inputs=[user_input, doc_file, doc_type, chatbot],
+        outputs=[chatbot, user_input]
+    )
+
+    # clear_btn = gr.Button("Clear Chat")
+    # clear_btn.click(lambda: [], outputs=[chatbot])
+
+    # with gr.Tab("📘 Domain Knowledge RAG"):
+    #     RAG_Domain_know_doc.demo.render()
+
+    # with gr.Tab("📁 User-Specific Documents"):
+    #     User_Specific_Documents.demo.render()
+
+    # with gr.Tab("🕰️ Old Documents"):
+    #     Old_Document.demo.render()
 
 if __name__ == "__main__":
     demo.launch(debug=True)
-    
+