Update graph_rag.py

graph_rag.py

@@ -5,8 +5,9 @@ import logging
 import subprocess
 from dotenv import load_dotenv
 import gradio as gr
+from utils import patient_info  # Importing patient_info from utils
 
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname=s' - '%(message)s')
 logger = logging.getLogger(__name__)
 
 # Load environment variables
@@ -68,7 +69,11 @@ def qa_tool_graph_rag(user_question):
         
         os.chdir(extract_path)
         
-        answer = run_graphrag_query(user_question, output_dir)
+        # Combine patient_info with user_question
+        combined_input = f"{patient_info}\n\n{user_question}"
+        
+        # Run the GraphRAG query with the combined input
+        answer = run_graphrag_query(combined_input, output_dir)
         
         logger.info(f"GraphRAG answer generated: {answer}")
         
@@ -84,5 +89,4 @@ def qa_tool_graph_rag(user_question):
         if 'output_dir' in locals():
             shutil.rmtree(output_dir)
         
-        os.chdir(original_dir)  # Return to the original directory
-
+        os.chdir(original_dir)