Update app.py

app.py

@@ -15,10 +15,10 @@ def qa_tool(user_question, method):
         logger.info(f"Regular RAG images: {images}")  # Additional log for debugging
         return answer, images, ask_another_visible, change_patient_visible, gr.update(visible=False)
     elif method == "Graph-RAG":
-        answer, images, ask_another_visible, change_patient_visible, graph_fig = qa_tool_graph_rag(user_question)
+        answer, images, ask_another_visible, change_patient_visible = qa_tool_graph_rag(user_question)
         logger.info("Graph-RAG answer generated")
         logger.info(f"Graph-RAG images: {images}")  # Additional log for debugging
-        return answer, images, ask_another_visible, change_patient_visible, gr.update(visible=True, value=graph_fig)
+        return answer, images, ask_another_visible, change_patient_visible, gr.update(visible=True)