Update app.py

app.py

@@ -167,201 +167,116 @@ if __name__ == "__main__":
 import gradio as gr
 import os
 import time
+import tempfile
 import requests
+from PIL import Image
+from io import BytesIO
 import re
 from datetime import datetime
 from dotenv import load_dotenv
-from huggingface_hub import HfApi, upload_file
-import uuid
 
+# Load environment variables
 load_dotenv()
 
-class BlogCreator:
-    def __init__(self):
-        self.hf_token = os.getenv("HF_TOKEN")
-        self.api = HfApi(token=self.hf_token)
-        self.headers = {"Authorization": f"Bearer {self.hf_token}"}
-        self.username = "jsakshi"  # Replace with your HF username
-
-    def generate_blog(self, topic):
-        try:
-            # Generate unique space name
-            space_name = f"{self.username}/blog-{uuid.uuid4().hex[:8]}"
-            
-            # Create blog space
-            self.api.create_repo(
-                repo_id=space_name,
-                repo_type="space",
-                space_sdk="static",
-                private=False
-            )
+# Hugging Face configuration
+HF_TOKEN = os.getenv("HF_TOKEN")
+TEXT_API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+IMAGE_API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-            # Generate content
-            blog_text, image_urls = self._generate_content(topic, space_name)
-            
-            # Create HTML with design
-            html_content = self._create_design(blog_text, image_urls, topic)
-            
-            # Upload files to space
-            self._upload_to_space(space_name, html_content, image_urls)
-            
-            return f"https://huggingface.co/spaces/{space_name}", blog_text
-            
-        except Exception as e:
-            return f"Error: {str(e)}", ""
+# Generate blog content
+def generate_blog_content(topic, tone="professional", length="long"):
+    try:
+        current_date = datetime.now().strftime("%B %d, %Y")
+        prompt = f"""<s>[INST] Write a {tone} blog post about {topic} with:
+        - Title and subtitle
+        - Introduction with statistics
+        - 3-4 detailed sections with subheadings
+        - A conclusion with key takeaways
+        - Markdown formatting with colorful HTML & CSS
+        - Published date: {current_date}
+        - At least 1500 words [/INST]</s>"""
 
-    def _generate_content(self, topic, space_name):
-        # Generate text
-        text_response = requests.post(
-            "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
-            headers=self.headers,
-            json={
-                "inputs": f"Write a comprehensive blog about {topic} with markdown formatting and 3 image placeholders [IMAGE1], [IMAGE2], [IMAGE3]",
-                "parameters": {"max_length": 2000}
-            }
-        )
-        blog_text = text_response.json()[0]['generated_text']
+        payload = {
+            "inputs": prompt,
+            "parameters": {"max_new_tokens": 2048, "temperature": 0.7, "return_full_text": False}
+        }
 
-        # Generate and upload images
-        image_urls = []
-        for idx in range(3):
-            img_response = requests.post(
-                "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0",
-                headers=self.headers,
-                json={"inputs": f"Colorful modern digital art about {topic}, vibrant colors, abstract patterns, 4k resolution"}
-            )
-            image_path = f"image_{idx}.png"
-            upload_file(
-                path_or_fileobj=img_response.content,
-                path_in_repo=image_path,
-                repo_id=space_name,
-                repo_type="space"
-            )
-            image_urls.append(image_path)
-            time.sleep(1)
+        response = requests.post(TEXT_API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
+        return response.json()[0]['generated_text']
+    except Exception as e:
+        return f"Error generating content: {str(e)}"
 
-        return blog_text, image_urls
+# Generate feature image
+def generate_featured_image(topic):
+    try:
+        prompt = f"A vibrant, eye-catching blog cover image about {topic}, ultra HD, artistic"
+        payload = {"inputs": prompt, "parameters": {"height": 1024, "width": 1024, "num_inference_steps": 30}}
+        response = requests.post(IMAGE_API_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
+        image = Image.open(BytesIO(response.content))
+        temp_img = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        image.save(temp_img.name)
+        return temp_img.name
+    except Exception as e:
+        return None
 
-    def _create_design(self, content, image_urls, topic):
-        # Replace image placeholders
-        for idx, url in enumerate(image_urls):
-            content = content.replace(f"[IMAGE{idx+1}]", f'![Image {idx+1}]({url})')
+# Convert blog into a colorful web page
+def generate_webpage(blog_content, title, author):
+    html_template = f"""
+    <html>
+    <head>
+        <style>
+            body {{ font-family: Arial, sans-serif; background-color: #f5f5f5; padding: 20px; }}
+            .container {{ max-width: 800px; margin: auto; background: white; padding: 20px; border-radius: 10px; }}
+            h1 {{ color: #2c3e50; }}
+            h2, h3 {{ color: #2980b9; }}
+            p {{ color: #333; line-height: 1.6; }}
+        </style>
+    </head>
+    <body>
+        <div class='container'>
+            <h1>{title}</h1>
+            <h3>by {author}</h3>
+            {blog_content.replace("\n", "<br>")}
+        </div>
+    </body>
+    </html>
+    """
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".html", mode="w", encoding="utf-8")
+    temp_file.write(html_template)
+    temp_file.close()
+    return temp_file.name
 
-        # Create styled HTML
-        return f"""
-<!DOCTYPE html>
-<html>
-<head>
-    <title>{topic}</title>
-    <style>
-        :root {{
-            --primary: #ff6b6b;
-            --secondary: #4ecdc4;
-            --accent: #ffe66d;
-        }}
-        body {{
-            max-width: 800px;
-            margin: 0 auto;
-            padding: 40px;
-            font-family: 'Segoe UI', sans-serif;
-            background: linear-gradient(45deg, #f8f9fa, #ffffff);
-            line-height: 1.6;
-        }}
-        .header {{
-            text-align: center;
-            padding: 40px 0;
-            background: var(--primary);
-            color: white;
-            border-radius: 15px;
-            margin-bottom: 40px;
-            box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-        }}
-        img {{
-            width: 100%;
-            border-radius: 15px;
-            margin: 25px 0;
-            border: 3px solid var(--secondary);
-            transition: transform 0.3s ease;
-        }}
-        img:hover {{
-            transform: scale(1.02);
-        }}
-        h1, h2, h3 {{
-            color: var(--primary);
-            margin-top: 30px;
-        }}
-        blockquote {{
-            border-left: 4px solid var(--accent);
-            padding: 15px 30px;
-            margin: 20px 0;
-            background: #fff9e6;
-            border-radius: 4px;
-        }}
-        a {{
-            color: var(--secondary);
-            text-decoration: none;
-            font-weight: bold;
-        }}
-        .content {{
-            background: white;
-            padding: 30px;
-            border-radius: 15px;
-            box-shadow: 0 2px 15px rgba(0,0,0,0.1);
-        }}
-    </style>
-</head>
-<body>
-    <div class="header">
-        <h1>{topic}</h1>
-        <p>Generated on {datetime.now().strftime("%B %d, %Y")}</p>
-    </div>
-    <div class="content">
-        {markdown.markdown(content)}
-    </div>
-</body>
-</html>
-"""
+# Generate blog
 
-    def _upload_to_space(self, space_name, html_content, image_urls):
-        # Upload HTML
-        upload_file(
-            path_or_fileobj=html_content.encode(),
-            path_in_repo="index.html",
-            repo_id=space_name,
-            repo_type="space"
-        )
+def generate_blog(topic, tone, author_name):
+    status_updates = ["🚀 Generating blog..."]
+    blog_content = generate_blog_content(topic, tone)
+    title = re.search(r'^#\s+(.+)$', blog_content, re.MULTILINE).group(1).strip() if re.search(r'^#\s+(.+)$', blog_content, re.MULTILINE) else topic
+    status_updates.append("🖼️ Generating featured image...")
+    image_path = generate_featured_image(topic)
+    status_updates.append("🌍 Generating website...")
+    web_page = generate_webpage(blog_content, title, author_name)
+    status_updates.append("✅ Blog Ready!")
+    return blog_content, title, "\n".join(status_updates), web_page
 
-with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🎨 AI Blog Designer")
-    
+# Gradio UI
+with gr.Blocks() as app:
+    gr.Markdown("# 🌟 AI Blog Generator & Web Designer")
     with gr.Row():
         with gr.Column():
-            topic_input = gr.Textbox(label="Blog Topic", placeholder="Enter your topic...")
-            generate_btn = gr.Button("Create Beautiful Blog", variant="primary")
-            
+            topic_input = gr.Textbox(label="Blog Topic")
+            tone_input = gr.Dropdown(["professional", "casual", "technical", "storytelling"], label="Writing Style", value="professional")
+            author_input = gr.Textbox(label="Author Name")
+            generate_btn = gr.Button("Generate Blog")
         with gr.Column():
-            gr.Markdown("## Your Blog Link")
-            blog_link = gr.Markdown("Waiting for generation...")
-            blog_preview = gr.HTML()
-            status = gr.Textbox(label="Status")
-
-    def process_request(topic):
-        creator = BlogCreator()
-        try:
-            url, content = creator.generate_blog(topic)
-            return (
-                f"[View Beautiful Blog]({url})", 
-                f"<iframe src='{url}' style='width:100%; height:600px; border:none;'></iframe>",
-                "✅ Blog created successfully!"
-            )
-        except Exception as e:
-            return "❌ Error", "", str(e)
-
-    generate_btn.click(
-        fn=process_request,
-        inputs=topic_input,
-        outputs=[blog_link, blog_preview, status]
-    )
+            title_output = gr.Textbox(label="Generated Title")
+            blog_output = gr.Markdown()
+            status_output = gr.Textbox(label="Status")
+            web_page_output = gr.File(label="Download Website")
+    generate_btn.click(generate_blog, inputs=[topic_input, tone_input, author_input], outputs=[blog_output, title_output, status_output, web_page_output])
 
 if __name__ == "__main__":
-    app.launch(share=True)
+    app.launch(share=True)