Update app.py

app.py

@@ -25,10 +25,12 @@ def extract_main_content(html_text):
 # ---------------------------
 # Prompt builder
 # ---------------------------
-def build_prompt(text, tones, platform):
+def build_prompt(text, tones, platform, blog_link):
     try:
         tone_str = ", ".join(tones)
         print(f"[build_prompt] Using tones: {tone_str} for platform: {platform}")
+        
+        # Add a friendly invitation to the end of the generated post
         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.
@@ -38,10 +40,7 @@ Make it {tone_str.lower()}, natural, and relatable. Include a catchy title to ho
 {text}
 ---
 
-Return in this format:
-Title: <catchy title>
-Body:
-<post body text>
+And by the way, if you're curious about my fermentation experiments or want to join in on the fun, feel free to check out my blog! I've got all sorts of tips, tricks, and beginner-friendly recipes to help you start your own fermentation journey. {blog_link} – hope to see you there!
 """
     except Exception as e:
         print(f"[build_prompt] Error: {e}")
@@ -50,23 +49,10 @@ Body:
 # ---------------------------
 # Function to call Gemini
 # ---------------------------
-def generate_post(api_key, text, tones, platform, humanize_output=False):
+def generate_post(api_key, text, tones, platform, blog_link):
     try:
         headers = {"Content-Type": "application/json"}
-        prompt = build_prompt(text, tones, platform)
-
-        if humanize_output:
-            prompt += """
-
-After generating the post, revise it to sound more naturally human and casual, as if written by someone genuinely sharing their experience with a bit of humor and self-deprecation.
-Soften the overly polished tone. Add minor imperfections (like fragments, casual slang, hesitations, etc.) to make it feel spontaneous.
-
-Return format:
-Title: <hooky title>
-Body:
-<revised post>
-"""
-
+        prompt = build_prompt(text, tones, platform, blog_link)
         data = {
             "contents": [{"parts": [{"text": prompt}]}]
         }
@@ -87,7 +73,7 @@ Body:
 # ---------------------------
 # Gradio UI
 # ---------------------------
-def app_interface(html_file, tones, platform_choice, api_key, humanize_output):
+def app_interface(html_file, tones, platform_choice, api_key, blog_link):
     try:
         if not html_file:
             return "Please upload an HTML blog post."
@@ -106,7 +92,7 @@ def app_interface(html_file, tones, platform_choice, api_key, humanize_output):
             return "Couldn't extract content from file. Ensure it's a valid HTML blog post."
 
         print("[app_interface] Generating post...")
-        output = generate_post(api_key, main_text, tones, platform_choice, humanize_output)
+        output = generate_post(api_key, main_text, tones, platform_choice, blog_link)
         return output
     except Exception as e:
         print(f"[app_interface] Error: {e}")
@@ -123,11 +109,11 @@ with gr.Blocks() as demo:
     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")
     api_key_input = gr.Textbox(label="Enter Your Gemini API Key", type="password")
-    humanizer_checkbox = gr.Checkbox(label="💬 Make More Human + Casual?", value=True)
+    blog_link_input = gr.Textbox(label="Enter Your Blog Link", placeholder="Insert your blog link here...")
     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, api_key_input, humanizer_checkbox], outputs=output_box)
+    generate_button.click(app_interface, inputs=[html_input, tone_checkboxes, platform_radio, api_key_input, blog_link_input], outputs=output_box)
 
 # ---------------------------
 # Launch App