Update app.py

app.py

@@ -48,7 +48,7 @@ def save_theme_to_dataset(xml_file):
             token=HF_TOKEN,
             commit_message="Updated website theme XML"
         )
-        return "✅ Theme successfully pushed to your private dataset!"
+        return "✅ Theme successfully pushed to your private dataset! Gemini will now use this for styling."
         
     except Exception as e:
         return f"❌ Error saving theme to dataset: {e}"
@@ -129,22 +129,12 @@ def generate_blogger_html(app_query, custom_link, mod_features, theme_color, tel
         }
         schema_script = f'<script type="application/ld+json">\n[{json.dumps(schema_app)}, {json.dumps(schema_faq)}, {json.dumps(schema_sysreq)}]\n</script>'
 
-        # Fetch Theme XML if available
-        try:
-            downloaded_theme_path = hf_hub_download(repo_id=DATASET_REPO, filename=THEME_FILE_NAME, repo_type="dataset", token=HF_TOKEN)
-            theme_background = "#0f172a" 
-        except Exception:
-            theme_background = "#0f172a"
+        # Fetch basic background color if we want a baseline, AI will handle the rest later
+        theme_background = "#0f172a" 
 
-        # Final Base HTML Template Construction
+        # Final Base HTML Template Construction (Fallback style before AI applies XML exact styles)
         html_template = f"""
 {schema_script}
-<style>
-  @keyframes pulse {{ 0% {{ transform: scale(1); }} 50% {{ transform: scale(1.02); box-shadow: 0 0 30px {theme_color}80; }} 100% {{ transform: scale(1); }} }}
-  .lex-btn:hover {{ filter: brightness(1.2); }}
-  .app-post-wrapper * {{ box-sizing: border-box; }}
-</style>
-
 <div style="display: none;">
   <img src="{icon_url}" alt="Download {title} Mod APK" />
 </div>
@@ -165,7 +155,7 @@ def generate_blogger_html(app_query, custom_link, mod_features, theme_color, tel
   </div>
 
   <div style="text-align: center; margin-bottom: 35px; background: #020617; padding: 25px; border-radius: 16px; border: 1px solid rgba(255,255,255,0.02);">
-     <a href="{custom_link}" target="_blank" rel="nofollow noopener" class="lex-btn" style="display: inline-block; width: 100%; max-width: 400px; background: {theme_color}; color: #fff; text-decoration: none; padding: 16px; border-radius: 50px; font-weight: 800; font-size: 1.15rem; box-shadow: 0 0 25px {theme_color}80; transition: transform 0.3s ease; animation: pulse 2s infinite;">
+     <a href="{custom_link}" target="_blank" rel="nofollow noopener" class="lex-btn" style="display: inline-block; width: 100%; max-width: 400px; background: {theme_color}; color: #fff; text-decoration: none; padding: 16px; border-radius: 50px; font-weight: 800; font-size: 1.15rem; box-shadow: 0 0 25px {theme_color}80; transition: transform 0.3s ease;">
         <i class="fa-solid fa-cloud-arrow-down" style="margin-right: 8px;"></i> Download APK via Secure Link
      </a>
      <p style="font-size: 0.8rem; color: #64748b; margin-top: 15px; margin-bottom: 0;">
@@ -207,7 +197,7 @@ def generate_blogger_html(app_query, custom_link, mod_features, theme_color, tel
 </div>
 """
         word_count = str(len(html_template.split()))
-        keyword_density = str(html_template.lower().count(title.lower()[:10])) # Safer substring counting
+        keyword_density = str(html_template.lower().count(title.lower()[:10])) 
         
         with open("latest_draft.txt", "w", encoding="utf-8") as f:
             f.write(html_template)
@@ -218,10 +208,10 @@ def generate_blogger_html(app_query, custom_link, mod_features, theme_color, tel
         return f"Error generating HTML: {str(e)}", f"<b>Error occurred: {str(e)}</b>", "0", "0"
 
 # -----------------------------
-# GEMINI AI ENHANCER
+# GEMINI AI ENHANCER (PRO MODEL + XML FETCH)
 # -----------------------------
 def enhance_with_gemini(raw_html, style_name, state):
-    """Uses Google Gemini to restructure and theme the HTML while retaining functionality."""
+    """Downloads XML from Dataset, reads it, and uses Gemini 1.5 Pro to exact-match the styles."""
     if not state.get("logged_in") or (time.time() - state.get("login_time", 0) > 1800):
         return "Session expired.", "Session Expired.", "0", "0", gr.update(visible=False)
 
@@ -231,14 +221,39 @@ def enhance_with_gemini(raw_html, style_name, state):
     if not raw_html or raw_html.startswith("Error"):
         return raw_html, "<b>Please generate Base HTML first before enhancing.</b>", "0", "0", gr.update(visible=False)
 
+    # 1. Fetch the exact XML file from Hugging Face Dataset
+    xml_content = ""
+    try:
+        downloaded_theme_path = hf_hub_download(
+            repo_id=DATASET_REPO, 
+            filename=THEME_FILE_NAME, 
+            repo_type="dataset", 
+            token=HF_TOKEN
+        )
+        with open(downloaded_theme_path, "r", encoding="utf-8") as f:
+            xml_content = f.read()
+    except Exception as e:
+        print(f"Failed to fetch XML: {e}")
+        xml_content = "WARNING: Could not fetch website_theme.xml from dataset. Please apply generic high-quality styles based on the name."
+
+    # 2. Configure Gemini Pro (Best for massive context like full XML files)
     try:
         genai.configure(api_key=GEMINI_API_KEY)
-        model = genai.GenerativeModel('gemini-1.5-flash') 
+        # Using Gemini 1.5 Pro because it excels at parsing massive context (XML files) and strict coding rules.
+        model = genai.GenerativeModel('gemini-1.5-pro') 
 
         prompt = f"""
         You are the lead front-end developer for the 'Lexical Space' blog.
-        I have generated the following raw HTML for an app post. It contains accurate factual data, JSON-LD schemas, exact image links, download URLs, and video iframes.
-        Your task is to enhance and re-style this HTML to perfectly match the '{style_name}' theme.
+        Your task is to restyle a raw HTML app post to EXACTLY match the design system of the website.
+
+        Below is the raw source code of the website's theme XML. It contains the exact CSS definitions, classes, ID's, CSS variables (like colors, fonts), and HTML structures used for the site.
+        
+        <WEBSITE_THEME_XML>
+        {xml_content}
+        </WEBSITE_THEME_XML>
+
+        I want you to format the following Base HTML App Post to match the style known as: '{style_name}'.
+        Look through the <WEBSITE_THEME_XML> provided above. Find the CSS classes, inline styles, or component structures that correspond to '{style_name}'. Apply those exact classes, color hexes, and layout structures to the Base HTML. DO NOT invent styles—extract them from the XML.
 
         STRICT RULES:
         1. CRITICAL: DO NOT remove or alter the `<script type="application/ld+json">` blocks. They are critical for SEO.
@@ -246,11 +261,6 @@ def enhance_with_gemini(raw_html, style_name, state):
         3. DO NOT hallucinate or change the factual data (file size, version, rating, text content).
         4. ONLY output raw HTML. Do not wrap your response in ```html blocks. Do not add conversational intro/outro text.
 
-        Theme Guidelines for '{style_name}':
-        - If 'Lexical Standard': Deep space background (#0f172a), neon accents, glassmorphism filters, glowing borders, and pulse animations.
-        - If 'Lexical Minimalist': Cleaner, fewer borders, flat white/light-gray or pure black backgrounds, high readability, subtle drop shadows instead of neon glow.
-        - If 'Lexical Max Store': Replicate a premium mobile app store UI (like Google Play / App Store). Use grid layouts, highly rounded corners (24px), large action buttons, and emphasize the app icon and download statistics.
-
         Base HTML Code to Enhance:
         {raw_html}
         """
@@ -293,7 +303,7 @@ with gr.Blocks(title="Lexical Space Internal Tool", theme=gr.themes.Monochrome(p
     # --- MAIN APPLICATION DASHBOARD ---
     with gr.Group(visible=False) as main_app:
         gr.Markdown("# 🚀 Lexical Space V2 Post Generator")
-        gr.Markdown("Search the Play Store, inject mods, and generate SEO-optimized HTML for your Blogger post.")
+        gr.Markdown("Search the Play Store, inject mods, and generate SEO-optimized HTML matching your XML theme.")
         
         with gr.Row():
             # --- LEFT COLUMN: SETTINGS ---
@@ -303,21 +313,22 @@ with gr.Blocks(title="Lexical Space Internal Tool", theme=gr.themes.Monochrome(p
                 link_input = gr.Textbox(label="Your Download Link (Mega, PixelDrain, etc.)", placeholder="https://...")
                 features_input = gr.Textbox(label="Mod Features (Comma separated)", placeholder="Premium Unlocked, No Ads", value="Premium Unlocked, No Ads, Analytics Removed")
                 
-                with gr.Accordion("Theme & XML Upload (Optional)", open=False):
-                    xml_theme_input = gr.File(label="Upload Website Theme XML")
+                with gr.Accordion("Theme & XML Upload", open=False):
+                    gr.Markdown("*Upload your Blogger XML theme here. It is saved to your private Hugging Face Dataset and read by Gemini for exact styling.*")
+                    xml_theme_input = gr.File(label="Upload website_theme.xml")
                     save_theme_btn = gr.Button("Save Theme to Dataset")
                     theme_status_out = gr.Textbox(label="Theme Status", interactive=False)
 
-                with gr.Accordion("Advanced SEO & Theme Settings", open=False):
-                    theme_color = gr.ColorPicker(label="Post Accent Color (Applied from Theme)", value="#3b82f6")
-                    telegram_link = gr.Textbox(label="Your Telegram Channel Link", value="https://t.me/lexicalspace") # Fixed raw URL mapping
+                with gr.Accordion("Advanced SEO Settings", open=False):
+                    theme_color = gr.ColorPicker(label="Fallback Accent Color", value="#3b82f6")
+                    telegram_link = gr.Textbox(label="Your Telegram Channel Link", value="https://t.me/lexicalspace") 
                     custom_size = gr.Textbox(label="Custom File Size Override (e.g. 150 MB)", placeholder="Optional Override...")
                     custom_date = gr.Textbox(label="Custom Updated Date Override (e.g. January 15, 2026)", placeholder="Optional Override...")
                 
                 generate_btn = gr.Button("⚙️ Generate Base HTML", variant="primary")
                 
                 gr.Markdown("<hr>")
-                gr.Markdown("### 2. AI Style Enhancer (Gemini)")
+                gr.Markdown("### 2. AI Style Enhancer (Gemini Pro)")
                 with gr.Row():
                     style_dropdown = gr.Dropdown(
                         choices=["Lexical Standard", "Lexical Minimalist", "Lexical Max Store"], 
@@ -325,7 +336,7 @@ with gr.Blocks(title="Lexical Space Internal Tool", theme=gr.themes.Monochrome(p
                         value="Lexical Standard",
                         scale=2
                     )
-                    enhance_btn = gr.Button("✨ Enhance Code", variant="secondary", scale=1)
+                    enhance_btn = gr.Button("✨ Enhance with XML Styles", variant="secondary", scale=1)
                 
                 gr.Markdown("<hr>")
                 gr.Markdown("### Post Statistics")