Update app.py

app.py

@@ -1,86 +1,52 @@
 import gradio as gr
-import subprocess
-import json
-import tempfile
-
-def generate_titles(api_key, blog_title):
-    import subprocess
-    import json
-    import tempfile
+import google.generativeai as genai
 
+# Function to generate Pinterest-optimized post titles
+def generate_pinterest_titles(api_key, blog_title):
     if not api_key or not blog_title:
-        return "Please provide both API key and blog title."
+        return "Please provide both your API key and blog title."
 
-    prompt = f"""
-You are a Pinterest SEO and viral marketing expert.
+    try:
+        # Configure Gemini API
+        genai.configure(api_key=api_key)
+        model = genai.GenerativeModel('gemini-1.5-pro-latest')
 
-Your task is to take this blog title:
-"{blog_title}"
+        # Prompt to inject emotion and make titles clickable for Pinterest
+        prompt = f"""
+You are a Pinterest SEO and viral marketing expert in the fermentation niche (kefir, kombucha, fermented veggies).
 
-And generate 5 viral, emotionally engaging, human-sounding Pinterest post titles.
+Your task is to transform this blog title:
+"{blog_title}"
 
+Into 5 emotionally engaging, curiosity-driven, and human-sounding Pinterest post titles. 
 These should:
-- Be optimized for Pinterest clicks
-- Include CTA (Call-To-Action)
-- Sound human, not robotic or like AI
-- Use emotional tone (excitement, curiosity, nostalgia, empowerment, etc.)
-- Focus on the fermentation niche (kefir, kombucha, sauerkraut, gut health, etc.)
-- Use no hashtags or emojis
-- Be short and highly clickable
-
-Return only the 5 titles, one per line.
+- Sound like a real human wrote them
+- Include strong CTA (call to action)
+- Spark emotion or curiosity (excitement, nostalgia, gut health benefits, etc.)
+- Avoid robotic or AI-sounding phrasing
+- Be highly clickable and engaging on Pinterest
+- No hashtags or emojis
+
+Return only the 5 titles, each on a separate line.
 """
 
-    with tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix=".json") as f:
-        json.dump({
-            "contents": [{"role": "user", "parts": [{"text": prompt}]}],
-            "generationConfig": {
-                "temperature": 0.9,
-                "topP": 1,
-                "topK": 1,
-                "maxOutputTokens": 256
-            }
-        }, f)
-        f.flush()
-
-        command = [
-            "curl",
-            "-s",
-            "-X", "POST",
-            "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro:generateContent",
-            "-H", f"Authorization: Bearer {api_key}",
-            "-H", "Content-Type: application/json",
-            "--data-binary", f"@{f.name}"
-        ]
-
-        try:
-            result = subprocess.run(command, capture_output=True, text=True, check=True)
-            response = json.loads(result.stdout)
-
-            # Check if candidates exist
-            if "candidates" in response:
-                return response["candidates"][0]["content"]["parts"][0]["text"].strip()
-            elif "error" in response:
-                return f"API Error: {response['error'].get('message', 'Unknown error')}"
-            else:
-                return f"Unexpected API Response:\n{json.dumps(response, indent=2)}"
-
-        except subprocess.CalledProcessError as e:
-            return f"API request failed:\n{e.stderr}"
-        except Exception as e:
-            return f"Exception occurred: {e}"
+        response = model.generate_content(prompt)
+        return response.text.strip()
 
+    except Exception as e:
+        return f"Error: {e}"
 
 # Gradio UI
-with gr.Blocks(title="Pinterest Title Generator - Fermentation Niche") as app:
-    gr.Markdown("## 🍓 Pinterest Title Generator for Fermentation Blogs\nTurn your blog title into viral Pinterest titles that drive clicks!")
-
-    api_key_input = gr.Textbox(label="🔑 Gemini API Key", placeholder="Paste your Gemini API key here", type="password")
-    blog_title_input = gr.Textbox(label="📝 Blog Title", placeholder="Enter your fermentation blog title")
-    submit_btn = gr.Button("🚀 Generate Pinterest Titles")
-
-    output_box = gr.Textbox(label="🎯 Pinterest Post Title Ideas", lines=10)
-
-    submit_btn.click(fn=generate_titles, inputs=[api_key_input, blog_title_input], outputs=output_box)
-
-app.launch()
+demo = gr.Interface(
+    fn=generate_pinterest_titles,
+    inputs=[
+        gr.Textbox(label="Gemini API Key", placeholder="Paste your Gemini API key here", type="password"),
+        gr.Textbox(label="Blog Title", placeholder="e.g., 10 Surprising Benefits of Drinking Kefir Every Day")
+    ],
+    outputs=gr.Textbox(label="Pinterest Post Titles (Humanized with CTA)", lines=8),
+    title="Pinterest Title Generator for Fermentation Blogs",
+    description="Generate humanized, emotional, CTA-packed Pinterest titles from your blog post titles using Gemini 1.5 Pro."
+)
+
+if __name__ == "__main__":
+    demo.launch()