Spaces:

jsakshi
/

BlogAgent

Sleeping

App Files Files Community

jsakshi commited on Feb 24, 2025

Commit

7c88afd

verified ·

1 Parent(s): 3d3e23c

Update app.py

Browse files

Files changed (1) hide show

app.py +141 -92

app.py CHANGED Viewed

@@ -171,7 +171,8 @@ import requests
 import re
 from datetime import datetime
 from dotenv import load_dotenv
-from huggingface_hub import HfApi, Repository
 load_dotenv()
@@ -180,21 +181,38 @@ class BlogCreator:
         self.hf_token = os.getenv("HF_TOKEN")
         self.api = HfApi(token=self.hf_token)
         self.headers = {"Authorization": f"Bearer {self.hf_token}"}
     def generate_blog(self, topic):
-        # Generate blog content
-        blog_text = self._generate_text(topic)
-        # Generate images
-        image_urls = self._generate_images(blog_text)
-        # Create hosted space
-        space_url = self._create_hosted_blog(topic, blog_text, image_urls)
-        return blog_text, space_url
-    def _generate_text(self, topic):
-        response = requests.post(
             "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
             headers=self.headers,
             json={
@@ -202,116 +220,147 @@ class BlogCreator:
                 "parameters": {"max_length": 2000}
             }
         )
-        return response.json()[0]['generated_text']
-    def _generate_images(self, content):
-        image_prompts = re.findall(r'\[IMAGE\d+\]', content)
         image_urls = []
-        for idx, _ in enumerate(image_prompts):
-            response = requests.post(
                 "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0",
                 headers=self.headers,
-                json={"inputs": f"Professional blog image about {content[:100]}, digital art, 4k"}
             )
-            img_url = f"https://huggingface.co/datasets/username/blogs/raw/main/image_{idx}.png"
-            self.api.upload_file(
-                path_or_fileobj=response.content,
-                path_in_repo=f"image_{idx}.png",
-                repo_id="username/blogs",
-                repo_type="dataset"
             )
-            image_urls.append(img_url)
             time.sleep(1)
-        return image_urls
-    def _create_hosted_blog(self, topic, content, image_urls):
-        # Create formatted HTML
-        repo_name = f"blog-{topic.lower().replace(' ', '-')}-{int(time.time())}"
-        html_content = f"""
-        <!DOCTYPE html>
-        <html>
-        <head>
-            <title>{topic}</title>
-            <style>
-                body {{ max-width: 800px; margin: 0 auto; padding: 20px; font-family: Arial, sans-serif; }}
-                img {{ width: 100%; border-radius: 10px; margin: 20px 0; }}
-                h1 {{ color: #2c3e50; }}
-                .header {{ text-align: center; padding: 40px 0; }}
-            </style>
-        </head>
-        <body>
-            <div class="header">
-                <h1>{topic}</h1>
-                <p>Generated on {datetime.now().strftime("%B %d, %Y")}</p>
-            </div>
-            {self._format_content(content, image_urls)}
-        </body>
-        </html>
-        """
-        # Create and host Space
-        self.api.create_repo(
-            repo_id=repo_name,
-            repo_type="space",
-            space_sdk="static"
-        )
-        self.api.upload_file(
             path_or_fileobj=html_content.encode(),
             path_in_repo="index.html",
-            repo_id=f"username/{repo_name}",
             repo_type="space"
         )
-        return f"https://huggingface.co/spaces/username/{repo_name}"
-    def _format_content(self, content, image_urls):
-        # Replace image placeholders with actual URLs
-        for idx, url in enumerate(image_urls):
-            content = content.replace(f"[IMAGE{idx+1}]", f'<img src="{url}" alt="Blog Image {idx+1}">')
-        # Convert markdown to HTML
-        content = content.replace("\n", "<br>")
-        content = re.sub(r"### (.*?)", r"<h3>\1</h3>", content)
-        content = re.sub(r"## (.*?)", r"<h2>\1</h2>", content)
-        content = re.sub(r"\*\*(.*?)\*\*", r"<strong>\1</strong>", content)
-        return content
-# Initialize blog creator
-creator = BlogCreator()
 with gr.Blocks(theme=gr.themes.Soft()) as app:
-    gr.Markdown("# 🌐 AI Blog Publisher")
     with gr.Row():
         with gr.Column():
             topic_input = gr.Textbox(label="Blog Topic", placeholder="Enter your topic...")
-            generate_btn = gr.Button("Generate & Publish", variant="primary")
         with gr.Column():
-            gr.Markdown("## Generated Content")
-            blog_preview = gr.Markdown()
-            space_link = gr.Markdown("**Hosted Blog URL:** Waiting for generation...")
             status = gr.Textbox(label="Status")
     def process_request(topic):
         try:
-            blog_text, url = creator.generate_blog(topic)
             return (
-                blog_text,
-                f"**Hosted Blog URL:** [Click to View]({url})",
-                "✅ Blog published successfully!"
             )
         except Exception as e:
-            return "", "❌ Error generating blog", str(e)
     generate_btn.click(
         fn=process_request,
         inputs=topic_input,
-        outputs=[blog_preview, space_link, status]
     )
 if __name__ == "__main__":

 import re
 from datetime import datetime
 from dotenv import load_dotenv
+from huggingface_hub import HfApi, upload_file
+import uuid
 load_dotenv()
         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
+            )
+            # 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)}", ""
+    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={
                 "parameters": {"max_length": 2000}
             }
         )
+        blog_text = text_response.json()[0]['generated_text']
+        # 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)
+        return blog_text, image_urls
+    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})')
+        # 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>
+"""
+    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"
         )
 with gr.Blocks(theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🎨 AI Blog 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")
         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]
     )
 if __name__ == "__main__":