Update app.py

app.py

@@ -49,6 +49,7 @@ Mimic top-tier PR publications using this structure:
 - Bullet lists for portfolio highlights or key achievements.
 - Insert **1 vision-driven quote** (real or fictional) if none provided.
 - Ready to paste into any CMS or press distribution tool.
+- Return ONLY the HTML content without any markdown code blocks or backticks.
 
 📥 INPUT FORMAT:
 User may provide raw facts, a bio, a business update, or a rough news draft.
@@ -59,10 +60,26 @@ User may provide raw facts, a bio, a business update, or a rough news draft.
   - ✅ Meta Title
   - ✅ Meta Description
   - ✅ Target Keyword
+- Do NOT wrap the output in ```html or ``` markdown blocks.
 
 Please transform the following content into a professional PR-style article:
 """
 
+    def clean_html_output(self, text):
+        """Remove markdown code blocks and clean HTML output"""
+        # Remove markdown code blocks (```html, ```HTML, ```, etc.)
+        cleaned = re.sub(r'```[a-zA-Z]*\n?', '', text)
+        cleaned = re.sub(r'\n?```', '', cleaned)
+        
+        # Remove any remaining backticks at start/end
+        cleaned = cleaned.strip('`')
+        
+        # Clean up extra whitespace
+        cleaned = re.sub(r'\n{3,}', '\n\n', cleaned)
+        cleaned = cleaned.strip()
+        
+        return cleaned
+
     def setup_gemini(self, api_key):
         """Configure Gemini API"""
         try:
@@ -189,7 +206,8 @@ Please transform the following content into a professional PR-style article:
             if not response.text:
                 return "❌ Error: No response generated from API", "", None
                 
-            article_html = response.text.strip()
+            # Clean the HTML output to remove markdown code blocks
+            article_html = self.clean_html_output(response.text.strip())
             
             # Create downloadable file
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -315,7 +333,7 @@ def create_interface():
         
         # Add examples
         gr.HTML("""
-        <div style="margin-top: 30px; padding: 20px; background-color: #f8f9fa; border-radius: 10px;">
+        <div style="margin-top: 30px; padding: 20px; border-radius: 10px;">
             <h3>💡 Example Input Content:</h3>
             <p><strong>Business Acquisition:</strong> "Tech entrepreneur John Smith announced the acquisition of downtown office building for $15M. Smith, who previously founded two successful startups, plans to convert the space into a modern co-working hub."</p>
             <p><strong>Product Launch:</strong> "AI company releases new software that reduces processing time by 60%. The tool has been tested by 500+ businesses with positive results."</p>
@@ -335,9 +353,4 @@ if __name__ == "__main__":
         os.system("pip install google-generativeai textstat")
     
     app = create_interface()
-    app.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True,
-        show_error=True
-    )
+    app.launch()