Update app.py

app.py

@@ -1,241 +1,99 @@
 import gradio as gr
-import os
-import google.generativeai as genai
 from bs4 import BeautifulSoup
-import datetime
-import re
-import html
+import json
+import requests
 
-# Function to extract text from HTML
-def extract_text_from_html(html_content):
-    soup = BeautifulSoup(html_content, 'html.parser')
-    # Remove script and style elements
-    for script in soup(["script", "style"]):
-        script.extract()
-    
-    # Get text
-    text = soup.get_text(separator='\n')
-    
-    # Remove extra whitespace
-    lines = (line.strip() for line in text.splitlines())
-    chunks = (phrase.strip() for line in lines for phrase in line.split("  "))
-    text = '\n'.join(chunk for chunk in chunks if chunk)
-    
+# ---------------------------
+# Gemini API Setup (User will insert their own key manually)
+# ---------------------------
+GEMINI_API_KEY = "YOUR_GEMINI_API_KEY_HERE"  # Replace manually
+GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro:generateContent"
+
+# ---------------------------
+# Tone Options
+# ---------------------------
+TONE_OPTIONS = ["Friendly", "Humanized", "Engaging", "Funny", "Emotional"]
+
+# ---------------------------
+# Function to extract main content from HTML
+# ---------------------------
+def extract_main_content(html_file):
+    soup = BeautifulSoup(html_file, "html.parser")
+    article = soup.find("article") or soup.body
+    text = article.get_text(separator="\n", strip=True) if article else ""
     return text
 
-# Generate post using Gemini API
-def generate_post(api_key, html_content, platform, styles, emotional_tone):
-    genai.configure(api_key=api_key)
-    model = genai.GenerativeModel('gemini-1.5-pro')
-    
-    # Extract text from HTML
-    content_text = extract_text_from_html(html_content)
-    
-    # Create a summary first (this helps control the context length)
-    summary_prompt = f"""
-    You are an expert content summarizer specializing in fermentation topics.
-    Please create a concise summary of the following content, focusing on the main points,
-    techniques, and unique aspects of the fermentation process described:
-    
-    {content_text[:8000]}  # Limiting to first 8000 chars for summary
-    """
-    
-    summary_response = model.generate_content(summary_prompt)
-    summary = summary_response.text
-    
-    # Prepare style instructions
-    style_instruction = ", ".join(styles)
-    if emotional_tone:
-        style_instruction += f" with a {emotional_tone} emotional tone"
-    
-    # Platform-specific formatting
-    if platform == "Reddit":
-        platform_instruction = "Format this as a Reddit post with a catchy title, engaging introduction, clear sections, and end with a question to encourage comments."
-    else:  # Quora
-        platform_instruction = "Format this as a Quora answer that demonstrates expertise, provides personal experience, includes detailed steps, and shows passion for fermentation."
-    
-    # Main generation prompt
-    prompt = f"""
-    You are an expert content creator specializing in fermentation topics.
-    
-    Create a {platform} post about fermentation based on the following content summary.
-    Use a {style_instruction} writing style.
-    
-    Write the post as if you have personally tried or experienced this fermentation technique/recipe.
-    Include sensory details (how it looked, tasted, smelled), challenges you faced, and results.
-    
-    The post should feel authentic, like it's written by a real fermentation enthusiast sharing their experience, not like marketing material.
-    
-    Content summary to base the post on:
-    {summary}
-    
-    Special instructions:
-    1. {platform_instruction}
-    2. Include a compelling title
-    3. Be specific and detailed about the fermentation process
-    4. Use conversational language that would engage the {platform} community
-    5. End with a question or call for the community to share their experiences
-    
-    Structure the response with:
-    TITLE: [Your generated title]
-    
-    [The body of your post]
-    """
-    
-    response = model.generate_content(prompt)
-    return response.text
+# ---------------------------
+# Prompt builder
+# ---------------------------
+def build_prompt(text, tones, platform):
+    tone_str = ", ".join(tones)
+    return f"""
+You are a skilled content writer. Based on the content below, generate a post for {platform}.
+The post should feel like someone sharing a personal experience or story related to the topic.
+Make it {tone_str.lower()}, natural, and relatable. Include a catchy title to hook readers, and a body that encourages responses.
 
-# Function to extract title and body from generated content
-def extract_title_and_body(generated_text):
-    # Find the title
-    title_match = re.search(r'TITLE:\s*(.*?)(?:\n\n|\n|$)', generated_text)
-    title = title_match.group(1) if title_match else "Generated Post"
-    
-    # Remove the title part to get the body
-    body = re.sub(r'TITLE:\s*.*?(?:\n\n|\n|$)', '', generated_text, count=1, flags=re.DOTALL).strip()
-    
-    return title, body
+---
+{text}
+---
 
-# Create formatted HTML for download
-def create_html_output(title, body):
-    return f"""<!DOCTYPE html>
-<html>
-<head>
-    <title>{html.escape(title)}</title>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <style>
-        body {{ font-family: Arial, sans-serif; line-height: 1.6; max-width: 800px; margin: 0 auto; padding: 20px; }}
-        h1 {{ color: #333; }}
-        .post-body {{ margin-top: 20px; }}
-        .post-meta {{ color: #666; font-size: 0.9em; margin-bottom: 20px; }}
-    </style>
-</head>
-<body>
-    <h1>{html.escape(title)}</h1>
-    <div class="post-meta">Generated on {datetime.datetime.now().strftime('%Y-%m-%d')} for {html.escape(platform_choice)}</div>
-    <div class="post-body">
-        {body.replace('\n', '<br>')}
-    </div>
-</body>
-</html>
+Return in this format:
+Title: <catchy title>
+Body:
+<post body text>
 """
 
-# Function to handle file upload and process request
-def process_upload(html_file, api_key, platform_choice, humanized, friendly, storytelling, emotional_tone):
+# ---------------------------
+# Function to call Gemini
+# ---------------------------
+def generate_post(text, tones, platform):
+    headers = {"Content-Type": "application/json"}
+    prompt = build_prompt(text, tones, platform)
+    data = {
+        "contents": [{"parts": [{"text": prompt}]}]
+    }
+    response = requests.post(
+        f"{GEMINI_API_URL}?key={GEMINI_API_KEY}",
+        headers=headers,
+        data=json.dumps(data)
+    )
+    result = response.json()
     try:
-        # Check if API key is provided
-        if not api_key.strip():
-            return "Please enter your Gemini API key", None, None, None, None
-        
-        # Read file content
-        html_content = html_file.decode('utf-8')
-        
-        # Collect selected styles
-        styles = []
-        if humanized:
-            styles.append("humanized")
-        if friendly:
-            styles.append("friendly")
-        if storytelling:
-            styles.append("personal storytelling")
-        
-        # Generate post
-        generated_post = generate_post(api_key, html_content, platform_choice, styles, emotional_tone)
-        
-        # Extract title and body
-        title, body = extract_title_and_body(generated_post)
-        
-        # Create formatted outputs for download
-        html_output = create_html_output(title, body)
-        text_output = f"{title}\n\n{body}"
-        
-        return f"## {title}\n\n{body}", title, body, html_output, text_output
-        
-    except Exception as e:
-        return f"Error: {str(e)}", None, None, None, None
+        return result["candidates"][0]["content"]["parts"][0]["text"]
+    except:
+        return "Error generating content. Check your API key or prompt."
+
+# ---------------------------
+# Gradio UI
+# ---------------------------
+def app_interface(html_file, tones, platform_choice):
+    if not html_file:
+        return "Please upload an HTML blog post."
+
+    html_content = html_file.read().decode("utf-8")
+    main_text = extract_main_content(html_content)
+    if not main_text:
+        return "Couldn't extract content from file. Ensure it's a valid HTML blog post."
+
+    output = generate_post(main_text, tones, platform_choice)
+    return output
+
+# Gradio Blocks UI
+with gr.Blocks() as demo:
+    gr.Markdown("# Reddit & Quora Post Generator 🧠")
+    gr.Markdown("Upload your HTML blog post and get ready-to-publish content for Reddit or Quora, written in a human, friendly tone.")
 
-# Define the Gradio interface
-with gr.Blocks(title="Fermentation Content Generator") as app:
-    gr.Markdown("# 🧪 Fermentation Content Generator for Reddit & Quora")
-    gr.Markdown("Upload your fermentation content HTML file and generate engaging posts optimized for Reddit or Quora.")
-    
     with gr.Row():
-        with gr.Column(scale=1):
-            # Input section
-            gr.Markdown("### 1. Configure Your Post")
-            
-            api_key = gr.Textbox(label="Gemini API Key", placeholder="Enter your API key", type="password")
-            
-            html_file = gr.File(label="Upload HTML Content", file_types=[".html", ".htm"])
-            
-            platform_choice = gr.Radio(
-                label="Choose Platform",
-                choices=["Reddit", "Quora"],
-                value="Reddit"
-            )
-            
-            with gr.Accordion("Style Options"):
-                gr.Markdown("Select the writing style:")
-                with gr.Row():
-                    humanized = gr.Checkbox(label="Humanized", value=True)
-                    friendly = gr.Checkbox(label="Friendly", value=True)
-                    storytelling = gr.Checkbox(label="Personal/Storytelling", value=True)
-                
-                emotional_tone = gr.Dropdown(
-                    label="Emotional Tone",
-                    choices=["Curious", "Enthusiastic", "Skeptical", "Inspiring", "Helpful", "None"],
-                    value="Enthusiastic"
-                )
-            
-            generate_btn = gr.Button("Generate Post", variant="primary")
-        
-        with gr.Column(scale=2):
-            # Output section
-            gr.Markdown("### 2. Generated Post Preview")
-            
-            output = gr.Markdown(label="Generated Post")
-            
-            with gr.Accordion("Post Details", open=False):
-                title_output = gr.Textbox(label="Title")
-                body_output = gr.Textbox(label="Body", lines=10)
-            
-            with gr.Row():
-                html_download = gr.File(label="Download as HTML")
-                text_download = gr.File(label="Download as Text")
-    
-    # Process when button is clicked
-    generate_btn.click(
-        fn=process_upload,
-        inputs=[html_file, api_key, platform_choice, humanized, friendly, storytelling, emotional_tone],
-        outputs=[output, title_output, body_output, html_download, text_download]
-    )
-    
-    # Examples and tips
-    with gr.Accordion("Tips & Examples", open=False):
-        gr.Markdown("""
-        ### Tips for Best Results
-        
-        1. **Content Quality**: Ensure your HTML file contains comprehensive information about the fermentation process.
-        2. **File Size**: For optimal results, keep your HTML file under 100KB.
-        3. **Style Selection**: Experiment with different style combinations to find what works best for your target audience.
-        4. **Emotional Tones**:
-           - **Curious**: Good for experimental fermentation techniques
-           - **Enthusiastic**: Works well for successful recipes and projects
-           - **Skeptical**: Useful for discussing common myths or challenges
-           - **Inspiring**: Great for showcasing transformative projects or health benefits
-           - **Helpful**: Best for educational content and guides
-        
-        ### Example Content
-        
-        Your HTML content should ideally include:
-        - Detailed fermentation process
-        - Ingredients and equipment used
-        - Timeline and observations
-        - Results and flavor notes
-        - Any troubleshooting tips
-        """)
+        html_input = gr.File(label="Upload HTML Blog Post", file_types=[".html"])
+
+    tone_checkboxes = gr.CheckboxGroup(label="Choose Tones", choices=TONE_OPTIONS, value=["Friendly", "Humanized"])
+    platform_radio = gr.Radio(label="Target Platform", choices=["Reddit", "Quora"], value="Reddit")
+    generate_button = gr.Button("Generate Post")
+    output_box = gr.Textbox(label="Generated Post", lines=20, interactive=True)
+
+    generate_button.click(app_interface, inputs=[html_input, tone_checkboxes, platform_radio], outputs=output_box)
 
-# Launch the app
-if __name__ == "__main__":
-    app.launch(share=True)
+# ---------------------------
+# Launch App
+# ---------------------------
+demo.launch()