Update app.py

app.py

@@ -45,7 +45,7 @@ phi_agent = create_phi_agent()
 
 # Response Generation
 def retrieve_and_generate_response(query, cross_encoder_choice, history=None):
-    print('hitry ', history)
+    print('history:', history)
     top_rerank = 25
     top_k_rank = 20
     
@@ -55,7 +55,7 @@ def retrieve_and_generate_response(query, cross_encoder_choice, history=None):
     try:
         start_time = perf_counter()
         query_vec = retriever.encode(query)
-        print('query vector ', query_vec)
+        print('query vector:', query_vec)
         documents = table.search(query_vec, vector_column_name="vector").limit(top_rerank).to_list()
         documents = [doc["text"] for doc in documents]
         
@@ -86,7 +86,22 @@ def retrieve_and_generate_response(query, cross_encoder_choice, history=None):
         logger.error(f"Error in response generation: {e}")
         return f"Error: {str(e)}"
 
-# Gradio Interface with Default Chatbot
+# Fixed chatbot function
+def chatbot_func(query, cross_encoder, history):
+    print('function called')
+    if not query.strip():
+        gr.Warning("Please submit a non-empty question")
+        return "", history
+    
+    response = retrieve_and_generate_response(query, cross_encoder, history)
+    history.append([query, ""])
+    
+    for character in response:
+        history[-1][1] += character
+        time.sleep(0.02)  # Streaming effect
+        yield "", history
+
+# Gradio Interface with Fixed Implementation
 with gr.Blocks(title="Science Chatbot", theme='gradio/soft') as chatbot_app:
     with gr.Row():
         with gr.Column(scale=10):
@@ -106,18 +121,13 @@ with gr.Blocks(title="Science Chatbot", theme='gradio/soft') as chatbot_app:
         with gr.Column(scale=3):
             gr.Image(value='logo.png', height=200, width=200, show_label=False)
     
-    # chatbot = gr.Chatbot(
-    #     [],
-    #     elem_id="chatbot",
-    #     avatar_images=(
-    #         'https://aui.atlassian.com/aui/8.8/docs/images/avatar-person.svg',
-    #         'https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo.svg'
-    #     ),
-    #     bubble_full_width=False,
-        
-    #     label="Conversation"
-    # )
-    chatbot=gr.Chatbot()
+    # Simple chatbot initialization
+    chatbot = gr.Chatbot(
+        value=[],  # Initialize with empty list
+        elem_id="chatbot",
+        bubble_full_width=False,
+        label="Conversation"
+    )
     
     with gr.Row():
         query = gr.Textbox(
@@ -136,32 +146,27 @@ with gr.Blocks(title="Science Chatbot", theme='gradio/soft') as chatbot_app:
     )
     clear = gr.Button("Clear")
     
-    def chatbot_func(query, cross_encoder, history=[]):
-        print('function called')
-        if not query.strip():
-            gr.Warning("Please submit a non-empty question")
-            return "", history
-        response = retrieve_and_generate_response(query, cross_encoder, history)
-        history.append((query, ""))
-        for character in response:
-            history[-1] = (query, history[-1][1] + character)
-            time.sleep(0.02)  # Streaming effect
-            yield "", history
-        return "", history
-    
+    # Fixed event handlers
     submit_btn.click(
-        chatbot_func,[query, cross_encoder, chatbot],[query, chatbot],
-        #queue=False
+        fn=chatbot_func,
+        inputs=[query, cross_encoder, chatbot],
+        outputs=[query, chatbot]
+    )
+    
+    query.submit(
+        fn=chatbot_func,
+        inputs=[query, cross_encoder, chatbot],
+        outputs=[query, chatbot]
+    )
+    
+    def clear_conversation():
+        return [], ""
+    
+    clear.click(
+        fn=clear_conversation,
+        inputs=None,
+        outputs=[chatbot, query]
     )
-    # query.submit(
-    #     chatbot_func,
-    #     inputs=[query, cross_encoder, chatbot],
-    #     outputs=[query, chatbot],
-    #     queue=False
-    # )
-    # clear.click(
-    #     lambda: ([], ""), None, [chatbot, query], queue=False
-    # )
     
     examples = [
         'CAN U SAY THE DIFFERENCES BETWEEN METALS AND NON METALS?',