Update app.py

app.py

@@ -1,97 +1,241 @@
-import gradio as gr import html2text import google.generativeai as genai
-
-Insert your Gemini API key manually
-
-GEMINI_API_KEY = "your-gemini-api-key-here" genai.configure(api_key=GEMINI_API_KEY)
-
-model = genai.GenerativeModel("gemini-1.5-pro-latest")
-
-Example HTML input (for demo/testing)
-
-EXAMPLE_HTML = """
-
+import gradio as gr
+import os
+import google.generativeai as genai
+from bs4 import BeautifulSoup
+import datetime
+import re
+import html
+
+# 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)
+    
+    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
+
+# 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
+
+# Create formatted HTML for download
+def create_html_output(title, body):
+    return f"""<!DOCTYPE html>
 <html>
-<head><title>How I Brew Kombucha at Home</title></head>
+<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>How I Brew Kombucha at Home</h1>
-<p>Brewing kombucha at home has changed my wellness journey. I start with black tea and sugar, let it cool, and then introduce the SCOBY. After 7 days, I bottle it with fresh ginger or berries for a second ferment. It's fizzy, healthy, and deeply satisfying.</p>
-<h2>Ingredients</h2>
-<ul>
-<li>1 SCOBY</li>
-<li>1 cup sugar</li>
-<li>8 cups filtered water</li>
-<li>4 bags black tea</li>
-</ul>
+    <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>
-"""Humanized prompt builder
-
-def build_prompt(html_content, platform, styles, emotions): markdown = html2text.html2text(html_content) prompt = f""" You are an expert copywriter and SEO specialist writing for a Reddit or Quora audience. Rewrite the following content into a post that feels genuinely human and experience-based.
-
-Platform: {platform}
-
-Writing style: {', '.join(styles)}
-
-Emotional tone: {', '.join(emotions)}
-
-Format as a personal story or real experience, not as polished marketing copy.
-
-Include imperfections, uncertainty, and real-life observations when possible.
-
-Use natural language and conversational tone. Avoid generic AI-sounding phrases.
-
-Add light humor, personal opinions, and authentic transitions.
-
-End with a relatable thought or casual call to action (like "What do you think?" or "Curious if anyone else has tried this?").
-
-Summarize where needed but preserve the heart of the experience.
-
-Output: A suggested title + post body. No tags, no metadata.
-
-
-Content: """ prompt += markdown return prompt
-
-Generator function
-
-def generate_post(file, platform, styles, emotions): if file is None: return "Please upload an HTML file."
-
-content = file.read().decode("utf-8")
-prompt = build_prompt(content, platform, styles, emotions)
-response = model.generate_content(prompt)
-return response.text
-
-Download helper
-
-def save_file(text, file_type): ext = ".html" if file_type == "HTML" else ".txt" return (text, f"generated_post{ext}")
-
-Gradio UI
-
-def launch_app(): with gr.Blocks(title="Reddit/Quora Post Generator") as demo: gr.Markdown("# Reddit/Quora Post Generator\nGenerate engaging posts based on SEO HTML content.")
-
-with gr.Row():
-        file_input = gr.File(label="Upload your HTML file")
-
-    with gr.Row():
-        platform = gr.CheckboxGroup(["Reddit", "Quora"], label="Choose platform (one only)", value=["Reddit"])
-        style = gr.CheckboxGroup(["Friendly", "Humanized", "Personal/Storytelling"], label="Choose writing style")
-        emotions = gr.CheckboxGroup(["Curious", "Enthusiastic", "Skeptical", "Inspiring"], label="Emotional Tone")
-
-    generate_btn = gr.Button("Generate Post")
-    output_text = gr.Textbox(label="Generated Post", lines=15)
-
+"""
+
+# Function to handle file upload and process request
+def process_upload(html_file, api_key, platform_choice, humanized, friendly, storytelling, emotional_tone):
+    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
+
+# 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():
-        download_format = gr.Radio(["Text", "HTML"], value="Text", label="Download Format")
-        download_btn = gr.DownloadButton("Download", label="Download Post")
-
-    def generate_and_return(file, platform, style, emotions, file_type):
-        platform_selected = platform[0] if platform else "Reddit"
-        result = generate_post(file, platform_selected, style, emotions)
-        return result, save_file(result, file_type)
-
-    generate_btn.click(generate_and_return,
-                      inputs=[file_input, platform, style, emotions, download_format],
-                      outputs=[output_text, download_btn])
-
-demo.launch()
-
-launch_app()
+        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
+        """)
+
+# Launch the app
+if __name__ == "__main__":
+    app.launch(share=True)