Update app.py

app.py

@@ -15,7 +15,6 @@ summarizer = None
 nlp = None
 initialization_status = "Initializing..."  # Track initialization state
 
-
 # ---------------------------- Helper Functions ----------------------------
 
 def log_error(message: str):
@@ -27,6 +26,39 @@ def log_error(message: str):
     except:
         print("Couldn't write to error log file.")  #If logging fails, still print to console
 
+# ---------------------------- Language Model Loading ----------------------------
+
+def load_spacy_model(model_name="en_core_web_sm"):
+    """Loads the SpaCy language model, downloading it if necessary."""
+    global initialization_status  # To update the initialization status
+
+    try:
+        print(f"Attempting to load SpaCy model '{model_name}'...")
+        nlp_model = spacy.load(model_name)
+        print(f"Successfully loaded SpaCy model '{model_name}'.")
+        initialization_status += f"\nSpaCy model '{model_name}' loaded."
+        return nlp_model
+    except OSError:
+        print(f"SpaCy model '{model_name}' not found. Downloading...")
+        initialization_status += f"\nSpaCy model '{model_name}' not found. Downloading..."
+        try:
+            import subprocess
+            subprocess.check_call(["python", "-m", "spacy", "download", model_name])
+            nlp_model = spacy.load(model_name)
+            print(f"Successfully loaded SpaCy model '{model_name}' after downloading.")
+            initialization_status += f"\nSuccessfully loaded SpaCy model '{model_name}' after downloading."
+            return nlp_model
+
+        except Exception as e:
+            log_error(f"Failed to download or load SpaCy model '{model_name}': {e}")
+            initialization_status += f"\nFailed to download or load SpaCy model '{model_name}': {e}"
+            return None  # Indicate failure
+
+    except Exception as e:
+        log_error(f"Error loading SpaCy model '{model_name}': {e}")
+        initialization_status += f"\nError loading SpaCy model '{model_name}': {e}"
+        return None
+
 # ---------------------------- Tool Functions ----------------------------
 
 def search_pubmed(query: str) -> list:
@@ -109,11 +141,22 @@ def setup():
     initialization_status = "Initializing..."
     try:
         print("Initializing summarization pipeline...")
+        initialization_status += "\nInitializing summarization pipeline..."
         summarizer = pipeline("summarization", model=SUMMARIZATION_MODEL, token=HUGGINGFACE_API_TOKEN)
         print("Summarization pipeline initialized.")
+        initialization_status += "\nSummarization pipeline initialized."
+
         print("Loading SpaCy model...")
-        nlp = spacy.load(SPACY_MODEL)
+        initialization_status += "\nLoading SpaCy model..."
+        global nlp
+        nlp = load_spacy_model()  # Call the SpaCy loading function.
+        if nlp is None:
+            initialization_status += "\nSpaCy model failed to load. Check the error log."
+            return initialization_status
+
         print("SpaCy model loaded.")
+        initialization_status += "\nSpaCy model loaded."
+
         initialization_status = "MedAI Agent initialized successfully!"
         return initialization_status  # Return the status message
     except Exception as e:
@@ -125,20 +168,16 @@ def setup():
 
 def launch_gradio():
     """Launches the Gradio interface."""
-    global initialization_status #Allows the function to modify global variable
+    global initialization_status  # Allows the function to modify global variable
     with gr.Blocks() as iface:
         gr.Markdown("# MedAI: Medical Literature Review and Summarization")
-        status_display = gr.Textbox(value=initialization_status, interactive=False) # Displays initialization status
+        status_display = gr.Textbox(value=initialization_status, interactive=False)  # Displays initialization status
         query_input = gr.Textbox(lines=3, placeholder="Enter your medical query (e.g., 'new treatments for diabetes')...")
         submit_button = gr.Button("Submit")
         output_results = gr.Markdown()
 
         submit_button.click(medai_agent, inputs=query_input, outputs=output_results)
-        #The run of the agent will not change.
-
-        #The setup is running. The value of the text display will update based on this step.
-        setup_result = setup()
-        status_display.value = setup_result #update the status display.
+        status_display.value = setup()  # Set the status after running setup
 
     iface.launch()