Upload app.py

app.py

@@ -242,11 +242,11 @@ def initialize_system():
 def ask_question(question: str, assistant: CurriculumAssistant):
     """Ask a question and get answer with relevant pages"""
     if not assistant or not assistant.qa_chain:
-        return "Please initialize the system first.", "", ""
+        return "Please initialize the system first."
     
     try:
-        # Get answer from QA chain
-        answer = assistant.qa_chain.run(question)
+        # Get answer from QA chain using invoke instead of run
+        answer = assistant.qa_chain.invoke({"query": question})
         
         # Find relevant pages
         relevant_pages = assistant.find_relevant_pages(question)
@@ -262,13 +262,13 @@ def ask_question(question: str, assistant: CurriculumAssistant):
             page_info = "No specific pages found for this question."
         
         # Format the complete response
-        full_response = f"## Answer\n\n{answer}\n\n---\n\n{page_info}"
+        full_response = f"## Answer\n\n{answer['result']}\n\n---\n\n{page_info}"
         
-        return full_response, answer, page_info
+        return full_response
         
     except Exception as e:
         error_msg = f"Error processing question: {str(e)}"
-        return error_msg, "", ""
+        return error_msg
 
 # Initialize the system
 status, assistant, curriculum_docs = initialize_system()
@@ -332,4 +332,4 @@ with gr.Blocks(title="Inclusive World Curriculum Assistant", theme=gr.themes.Sof
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()