Update app.py

app.py

@@ -8,11 +8,10 @@ from langchain_core.messages import HumanMessage
 import os
 import re
 
-# --- Configuration & Initialization ---
-# For deployment on Hugging Face, set GEMINI_API_KEY in the Space's secrets
-llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", google_api_key=os.environ.get("GEMINI_API_KEY"))
+# --- Configuration ---
+# We no longer initialize the LLM here. We will do it on-demand.
 
-# --- Core Scraping & Analysis Functions ---
+# --- Core Scraping & Analysis Functions (Unchanged) ---
 
 def fetch_html(url):
     """Fetches HTML content from a URL with a browser-like user-agent."""
@@ -29,21 +28,12 @@ def analyze_onpage_seo(soup):
     """Analyzes the on-page SEO elements of a webpage."""
     title = soup.find('title').get_text(strip=True) if soup.find('title') else "N/A"
     description = soup.find('meta', attrs={'name': 'description'})['content'] if soup.find('meta', attrs={'name': 'description'}) else "N/A"
-    
     headings = {'h1': [], 'h2': [], 'h3': []}
     for h_tag in ['h1', 'h2', 'h3']:
         for tag in soup.find_all(h_tag):
             headings[h_tag].append(tag.get_text(strip=True))
-            
-    # Simple word count
     word_count = len(soup.get_text().split())
-    
-    return {
-        "title": title,
-        "description": description,
-        "headings": headings,
-        "word_count": word_count
-    }
+    return {"title": title, "description": description, "headings": headings, "word_count": word_count}
 
 def analyze_tech_stack(soup, html):
     """Performs a very basic tech stack analysis."""
@@ -55,36 +45,49 @@ def analyze_tech_stack(soup, html):
     if "gtag('config'" in html: tech.add("Google Analytics (GA4)")
     if "GTM-" in html: tech.add("Google Tag Manager")
     if soup.find('link', href=lambda x: x and 'tailwind' in x): tech.add("Tailwind CSS")
-        
     return list(tech) if tech else ["Basic HTML/CSS"]
 
+# --- AI Summary Generation (Now with Robust Initialization) ---
+
 def generate_ai_summary(url, seo_data, tech_data):
     """Generates a high-level strategic summary using an LLM."""
-    prompt = f"""
-    You are a world-class marketing strategist. Analyze the following data for the website {url} and provide a concise, actionable summary.
-
-    ## On-Page SEO Analysis:
-    - **Title:** {seo_data['title']}
-    - **Meta Description:** {seo_data['description']}
-    - **H1 Headings:** {', '.join(seo_data['headings']['h1'])}
-    - **Page Word Count:** {seo_data['word_count']}
-
-    ## Technology Stack:
-    - {', '.join(tech_data)}
-
-    ## Your Analysis:
-    Provide a 3-bullet point summary covering:
-    1.  **Their Primary Goal:** What is this page trying to achieve based on its language and structure?
-    2.  **Their Target Audience:** Who are they talking to?
-    3.  **A Key Strategic Insight:** What is one clever thing they are doing, or one major missed opportunity?
-    """
+    
+    # ** THE FIX: Lazy Initialization **
+    # We initialize the LLM inside the function, only when it's needed.
+    api_key = os.environ.get("GEMINI_API_KEY")
+    
+    # Provide a clear error message if the key is missing.
+    if not api_key:
+        return "ERROR: `GEMINI_API_KEY` is not set in the Space secrets. The AI summary cannot be generated. Please ask the Space owner to add it."
+
     try:
+        llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", google_api_key=api_key)
+        
+        prompt = f"""
+        You are a world-class marketing strategist. Analyze the following data for the website {url} and provide a concise, actionable summary.
+
+        ## On-Page SEO Analysis:
+        - **Title:** {seo_data['title']}
+        - **Meta Description:** {seo_data['description']}
+        - **H1 Headings:** {', '.join(seo_data['headings']['h1'])}
+        - **Page Word Count:** {seo_data['word_count']}
+
+        ## Technology Stack:
+        - {', '.join(tech_data)}
+
+        ## Your Analysis:
+        Provide a 3-bullet point summary covering:
+        1.  **Their Primary Goal:** What is this page trying to achieve based on its language and structure?
+        2.  **Their Target Audience:** Who are they talking to?
+        3.  **A Key Strategic Insight:** What is one clever thing they are doing, or one major missed opportunity?
+        """
         response = llm.invoke([HumanMessage(content=prompt)])
         return response.content
     except Exception as e:
-        return f"Could not generate AI summary: {e}"
+        # Provide a more specific error if the API call itself fails
+        return f"Could not generate AI summary. The API call failed: {e}"
 
-# --- The Main Orchestrator ---
+# --- The Main Orchestrator (Unchanged) ---
 
 def competitor_teardown(url):
     """The main function that runs the entire analysis pipeline."""
@@ -107,9 +110,6 @@ def competitor_teardown(url):
     yield "Generating AI Summary...", None, None, None, None, gr.Button("Analyzing...", interactive=False)
     ai_summary = generate_ai_summary(url, seo_data, tech_data)
 
-    # --- Prepare Outputs ---
-    
-    # SEO Output
     seo_md = f"""
     ### 👑 SEO & Content
     | Metric | Value |
@@ -123,12 +123,11 @@ def competitor_teardown(url):
     - **H2 Tags ({len(seo_data['headings']['h2'])}):** {len(seo_data['headings']['h2'])} found
     """
     
-    # Tech Stack Output
     tech_md = "### stack Tech Stack\n\n" + "\n".join([f"- `{t}`" for t in tech_data])
     
     yield ai_summary, seo_md, tech_md, "Analysis Complete", "More features coming soon!", gr.Button("Analyze", interactive=True)
 
-# --- Gradio UI ---
+# --- Gradio UI (Unchanged) ---
 with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important;}") as demo:
     gr.Markdown("# 🕵️ Gumbo Board: The Instant Competitor Teardown")
     gr.Markdown("Enter a competitor's website to get an instant analysis of their online strategy. *Powered by Gumbo (BeautifulSoup) & AI.*")
@@ -149,7 +148,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important;}")
         with gr.TabItem("Social Presence (Coming Soon)"):
             social_output = gr.Markdown()
 
-    # Link the button click to the main function
     submit_btn.click(
         fn=competitor_teardown,
         inputs=[url_input],
@@ -159,6 +157,4 @@ with gr.Blocks(theme=gr.themes.Soft(), css="footer {display: none !important;}")
     gr.Markdown("---")
     gr.Markdown("### Ready for More? \n Get unlimited reports, save projects, and export to PDF with our Pro plan. \n **[🚀 Launching Soon - Sign up on Gumroad!](https://gumroad.com/)**")
 
-
-# Launch the app
 demo.launch()