Upload app.py

app.py

@@ -133,6 +133,17 @@ def generate_response(user_input: str, vector_store: FAISS) -> Optional[str]:
         return "Une erreur s'est produite lors de la génération de la réponse."
 
 
+def chatbot(query):
+    """Main function to run the chatbot"""
+    try:
+        vs = initialize_vector_store()
+        response = generate_response(query, vs)
+        return response or "Aucune réponse générée."
+    except Exception as e:
+        logging.error(f"Chatbot error: {str(e)}")
+        return f"Une erreur s'est produite : {str(e)}"
+
+# Gradio Interface Simplification
 def get_random_questions():
     """Get 3 random example questions"""
     return random.sample(EXAMPLE_QUESTIONS, 3)
@@ -147,10 +158,10 @@ with gr.Blocks(title="MTC Chatbot", theme=gr.themes.Soft()) as demo:
     """)
     
     # Chat Interface
-    chatbot_ui = gr.Chatbot(
+    chatbot = gr.Chatbot(
         label="Dialogue",
-        type="chatbot"
-    )
+        bubble_full_width=False,
+        )
     
     # Input Section
     with gr.Row():
@@ -170,18 +181,6 @@ with gr.Blocks(title="MTC Chatbot", theme=gr.themes.Soft()) as demo:
             examples_per_page=3
         )
     
-    # Hidden API Endpoint
-    with gr.Row(visible=False):
-        api_input = gr.Textbox()
-        api_output = gr.Textbox()
-        api_btn = gr.Button()
-        api_btn.click(
-            fn=api_respond,
-            inputs=api_input,
-            outputs=api_output,
-            api_name="/respond"
-        )
-    
     # Chat Functions
     def respond(message, chat_history):
         try:
@@ -196,14 +195,15 @@ with gr.Blocks(title="MTC Chatbot", theme=gr.themes.Soft()) as demo:
     # Event Handling
     btn.click(
         respond,
-        [msg, chatbot_ui],
-        [msg, chatbot_ui]
+        [msg, chatbot],
+        [msg, chatbot]
     )
     msg.submit(
         respond,
-        [msg, chatbot_ui],
-        [msg, chatbot_ui]
+        [msg, chatbot],
+        [msg, chatbot]
     )
 
 if __name__ == "__main__":
     demo.launch()
+