Update app.py

app.py

@@ -144,10 +144,10 @@ def create_rag_chain(chat_history: str):
 chat_history = ChatHistory()
 
 # Gradio Function
-def ask_question_gradio(question, history: gr.Chatbot) -> Tuple[str, gr.Chatbot]:
+def ask_question_gradio(question, history):
     try:
         # Add user question to chat history
-        chat_history.add_message("User", question)
+        chat_history.add_message("user", question)
         
         # Get formatted history
         formatted_history = chat_history.get_formatted_history()
@@ -161,19 +161,20 @@ def ask_question_gradio(question, history: gr.Chatbot) -> Tuple[str, gr.Chatbot]
             response += chunk
         
         # Add assistant response to chat history
-        chat_history.add_message("Assistant", response)
+        chat_history.add_message("assistant", response)
         
         # Update Gradio chat history
-        history.append((question, response))
+        history.append({"role": "user", "content": question})
+        history.append({"role": "assistant", "content": response})
         
         return "", history
     except Exception as e:
         logger.error(f"Error during question processing: {e}")
-        return "", history + [("Error", "An error occurred. Please try again later.")]
+        return "", history + [{"role": "assistant", "content": "An error occurred. Please try again later."}]
 
 def clear_chat():
     chat_history.clear()
-    return None
+    return [], ""
 
 # Gradio Interface
 with gr.Blocks() as iface:
@@ -183,6 +184,7 @@ with gr.Blocks() as iface:
     chatbot = gr.Chatbot(
         height=400,
         show_label=False,
+        type="messages"  # Using the new messages format
     )
     
     with gr.Row():
@@ -200,9 +202,8 @@ with gr.Blocks() as iface:
     )
     
     clear_button.click(
-        lambda: (None, None),
-        outputs=[chatbot, question_input],
-        _js="() => { window.location.reload(); }"
+        clear_chat,
+        outputs=[chatbot, question_input]
     )
 
 # Launch the Gradio App