Update app.py

app.py

@@ -3,11 +3,7 @@ import html2text
 import google.generativeai as genai
 import os
 
-# Configure Gemini API
-genai.configure(api_key="your-gemini-api-key-here")
-model = genai.GenerativeModel("gemini-pro")
-
-# Prompt template
+# Prompt template builder
 def build_prompt(content, platform, styles, emotions):
     style_text = ", ".join(styles) if styles else "humanized"
     emotion_text = ", ".join(emotions) if emotions else "genuine"
@@ -27,25 +23,45 @@ CONTENT TO CONVERT:
 \"\"\"{content}\"\"\"
 """
 
-# Function to handle generation
-def generate_post(file, platform, styles, emotions):
+# Function to handle post generation
+def generate_post(api_key, file, platform, styles, emotions):
+    if not api_key:
+        return "Please enter your Gemini API key first."
     if not file:
         return "Please upload a file."
 
-    # Read and convert HTML file
-    raw_html = file.read().decode("utf-8")
-    plain_text = html2text.html2text(raw_html)
+    try:
+        genai.configure(api_key=api_key)
+        model = genai.GenerativeModel("gemini-pro")
+
+        raw_html = file.read().decode("utf-8")
+        plain_text = html2text.html2text(raw_html)
+        prompt = build_prompt(plain_text, platform, styles, emotions)
+
+        response = model.generate_content(prompt)
+        return response.text
 
-    prompt = build_prompt(plain_text, platform, styles, emotions)
-    response = model.generate_content(prompt)
-    return response.text
+    except Exception as e:
+        return f"Error: {e}"
 
-# Gradio UI
+# Download handlers
+def download_outputs(content):
+    html_path = "/tmp/post_output.html"
+    txt_path = "/tmp/post_output.txt"
+    with open(html_path, "w") as f:
+        f.write(content)
+    with open(txt_path, "w") as f:
+        f.write(content)
+    return html_path, txt_path
+
+# Gradio App
 with gr.Blocks(title="Reddit/Quora Post Generator") as demo:
-    gr.Markdown("## 🔥 Turn your SEO content into humanized Reddit or Quora posts")
+    gr.Markdown("## 🔥 Turn your SEO-rich HTML content into humanized Reddit or Quora posts")
+
+    api_key = gr.Textbox(label="🔐 Gemini API Key", type="password", placeholder="Paste your Gemini API key")
 
     with gr.Row():
-        file_input = gr.File(label="Upload HTML file", file_types=[".html"])
+        file_input = gr.File(label="Upload SEO HTML file", file_types=[".html"])
         platform = gr.Dropdown(["Reddit", "Quora"], label="Select Platform", value="Reddit")
 
     with gr.Row():
@@ -60,23 +76,14 @@ with gr.Blocks(title="Reddit/Quora Post Generator") as demo:
             value=["Curious"]
         )
 
-    generate_btn = gr.Button("Generate Post")
+    generate_btn = gr.Button("✨ Generate Post")
     output = gr.Textbox(label="Generated Post", lines=15)
 
     with gr.Row():
-        download_html = gr.File(label="Download as .html", interactive=False)
-        download_txt = gr.File(label="Download as .txt", interactive=False)
-
-    def download_outputs(content):
-        html_path = "/tmp/post_output.html"
-        txt_path = "/tmp/post_output.txt"
-        with open(html_path, "w") as f:
-            f.write(content)
-        with open(txt_path, "w") as f:
-            f.write(content)
-        return html_path, txt_path
-
-    generate_btn.click(fn=generate_post, inputs=[file_input, platform, styles, emotions], outputs=output)
+        download_html = gr.File(label="⬇️ Download as .html", interactive=False)
+        download_txt = gr.File(label="⬇️ Download as .txt", interactive=False)
+
+    generate_btn.click(fn=generate_post, inputs=[api_key, file_input, platform, styles, emotions], outputs=output)
     output.change(fn=download_outputs, inputs=output, outputs=[download_html, download_txt])
 
 demo.launch()