Update gemini_report.py

gemini_report.py

@@ -1,13 +1,11 @@
 import json
 import google.generativeai as genai
-from config import GEMINI_API_KEY
-from models import PageSpeedRequest
 
-def generate_report_with_gemini(pagespeed_data: dict) -> str:
-    genai.configure(api_key=GEMINI_API_KEY)
+def generate_report_with_gemini(pagespeed_data, gemini_api_key):
+    genai.configure(api_key=gemini_api_key)
     model = genai.GenerativeModel("gemini-2.0-flash")
 
-    prompt = (
+        prompt = (
         "**Role:** You are an **Expert Web Performance Optimization Analyst and Senior Full-Stack Engineer** "
         "with deep expertise in interpreting Google PageSpeed Insights data, diagnosing frontend and "
         "backend bottlenecks, and devising actionable, high-impact optimization strategies.\n\n"
@@ -78,13 +76,13 @@ def generate_report_with_gemini(pagespeed_data: dict) -> str:
         "Avoid generic advice; all recommendations must be tied to specific findings within the report. "
         "Do not add anything irrelevant in the report.**"
     )
+
     try:
         response = model.generate_content(prompt)
         if response and hasattr(response, "text"):
             return response.text
-        elif response and getattr(response, "candidates", None):
-            if response.candidates[0].finish_reason == "SAFETY":
-                return "Report generation was blocked due to safety settings."
-        return "Report generation failed or returned an incomplete response."
+        elif response and response.candidates and response.candidates[0].finish_reason == "SAFETY":
+            return "Report generation was blocked due to safety settings."
+        return "Report generation resulted in a non-text response or was incomplete."
     except Exception as e:
-        return f"An error occurred during report generation: {str(e)}"
+        return f"An error occurred during report generation: {str(e)}."