Update app.py

app.py

@@ -467,6 +467,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
 if __name__ == "__main__":
     app.launch(share=True)'''
 
+
+
+           
 import gradio as gr
 import os
 import time
@@ -479,15 +482,22 @@ from dotenv import load_dotenv
 from huggingface_hub import HfApi, upload_file
 import json
 from functools import partial
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
 
 load_dotenv()
 
 # Configuration
 HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    logger.error("HF_TOKEN not found in .env file")
 HF_USERNAME = "jsakshi"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-# Simulated user authentication (in practice, use a proper auth system)
+# Simulated user authentication (replace with proper auth in production)
 AUTHORIZED_USERS = {"admin": "password123"}  # username: password
 
 # Store edit history for undo/redo
@@ -495,49 +505,79 @@ edit_history = []
 current_history_index = -1
 
 def generate_initial_content(topic):
-    # Same as original generate_blog_content but simplified for example
-    prompt = f"Create a blog post about {topic} with title, subtitle, introduction, 3 sections, and conclusion."
-    response = requests.post(
-        "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
-        headers=HEADERS,
-        json={"inputs": prompt, "parameters": {"max_length": 2000}}
-    )
-    if response.status_code == 200:
+    """Generate initial blog content using Hugging Face API."""
+    logger.info(f"Generating content for topic: {topic}")
+    try:
+        prompt = f"""Create a detailed blog post about {topic} including:
+        - A compelling title and subtitle
+        - An introduction
+        - 3 main sections with descriptive headings
+        - A conclusion
+        Use an informative tone."""
+        response = requests.post(
+            "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
+            headers=HEADERS,
+            json={"inputs": prompt, "parameters": {"max_length": 2000}}
+        )
+        if response.status_code != 200:
+            logger.error(f"API request failed: {response.status_code} - {response.text}")
+            return f"Error: API request failed with status {response.status_code}"
         return response.json()[0]['generated_text']
-    return f"# {topic}\n## Subtitle\nIntro text\n## Section 1\nText\n## Section 2\nText\n## Section 3\nText\n## Conclusion\nText"
+    except Exception as e:
+        logger.error(f"Error generating content: {str(e)}")
+        return f"Error generating content: {str(e)}"
+
+def generate_image(prompt):
+    """Generate an image using Stable Diffusion API."""
+    try:
+        response = requests.post(
+            "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0",
+            headers=HEADERS,
+            json={"inputs": prompt}
+        )
+        if response.status_code == 200:
+            return response.content
+        logger.error(f"Image generation failed: {response.status_code} - {response.text}")
+        return None
+    except Exception as e:
+        logger.error(f"Error generating image: {str(e)}")
+        return None
 
 def create_or_update_space(content_data, space_name=None, images=[]):
-    api = HfApi(token=HF_TOKEN)
-    
-    if not space_name:
-        space_id = f"blog-{uuid.uuid4().hex[:8]}"
-        space_name = f"{HF_USERNAME}/{space_id}"
-        api.create_repo(repo_id=space_name, repo_type="space", space_sdk="static", private=False)
-    
-    # Generate HTML with contenteditable attributes
-    sections = re.split(r'(## .+)', content_data)
-    html_content = '<div class="editable-container">'
-    current_section = ""
-    
-    for part in sections:
-        if part.strip():
-            if part.startswith('## '):
-                if current_section:
-                    html_content += f'<div class="section-content" contenteditable="true">{markdown.markdown(current_section)}</div></div>'
-                html_content += f'<div class="section"><h2 class="editable-header" contenteditable="true">{part[3:]}</h2>'
-                current_section = ""
-            else:
-                current_section += part
-    if current_section:
-        html_content += f'<div class="section-content" contenteditable="true">{markdown.markdown(current_section)}</div></div>'
-    html_content += '</div>'
-    
-    # Add images
-    for i, img in enumerate(images):
-        html_content += f'<div class="image-container" draggable="true" data-index="{i}"><img src="{img}" class="editable-image" /><button class="delete-image">Delete</button></div>'
-    
-    complete_html = f"""<!DOCTYPE html>
-<html>
+    """Create or update a Hugging Face Space with editable content."""
+    try:
+        api = HfApi(token=HF_TOKEN)
+        if not space_name:
+            space_id = f"blog-{uuid.uuid4().hex[:8]}"
+            space_name = f"{HF_USERNAME}/{space_id}"
+            api.create_repo(repo_id=space_name, repo_type="space", space_sdk="static", private=False)
+            logger.info(f"Created new space: {space_name}")
+
+        # Process content into editable sections
+        sections = re.split(r'(## .+)', content_data)
+        html_content = '<div class="editable-container">'
+        current_section = ""
+        for part in sections:
+            if part.strip():
+                if part.startswith('## '):
+                    if current_section:
+                        html_content += f'<div class="section-content" contenteditable="true">{markdown.markdown(current_section)}</div></div>'
+                    html_content += f'<div class="section"><h2 class="editable-header" contenteditable="true">{part[3:]}</h2>'
+                    current_section = ""
+                else:
+                    current_section += part
+        if current_section:
+            html_content += f'<div class="section-content" contenteditable="true">{markdown.markdown(current_section)}</div></div>'
+        html_content += '</div>'
+
+        # Add images
+        image_html = ""
+        for i, img_path in enumerate(images):
+            image_html += f'<div class="image-container" draggable="true" data-index="{i}"><img src="{img_path}" class="editable-image" alt="Blog image" /><button class="delete-image">Delete</button></div>'
+
+        # Complete HTML with editing features
+        complete_html = f"""<!DOCTYPE html>
+<html lang="en">
 <head>
     <meta charset="UTF-8">
     <title>Editable Blog</title>
@@ -548,8 +588,9 @@ def create_or_update_space(content_data, space_name=None, images=[]):
         .image-container {{ position: relative; margin: 20px 0; }}
         .editable-image {{ width: 100%; max-width: 500px; cursor: move; }}
         .delete-image {{ position: absolute; top: 5px; right: 5px; }}
-        .editing-tools {{ position: fixed; top: 10px; left: 10px; background: white; padding: 10px; border: 1px solid #ccc; }}
+        .editing-tools {{ position: fixed; top: 10px; left: 10px; background: white; padding: 10px; border: 1px solid #ccc; z-index: 1000; }}
         [contenteditable]:focus {{ outline: 2px solid #2D68C4; }}
+        body {{ font-family: Arial, sans-serif; line-height: 1.6; }}
     </style>
     <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
 </head>
@@ -569,6 +610,7 @@ def create_or_update_space(content_data, space_name=None, images=[]):
         <button onclick="redo()">Redo</button>
     </div>
     {html_content}
+    {image_html}
     <script>
         let currentSpace = "{space_name}";
         let images = {json.dumps(images)};
@@ -618,7 +660,7 @@ def create_or_update_space(content_data, space_name=None, images=[]):
             }}
         }}
         
-        // Drag and drop
+        // Drag and drop images
         document.querySelectorAll('.image-container').forEach(container => {{
             container.addEventListener('dragstart', e => {{
                 e.dataTransfer.setData('text/plain', container.dataset.index);
@@ -658,30 +700,76 @@ def create_or_update_space(content_data, space_name=None, images=[]):
     </script>
 </body>
 </html>"""
-    
-    upload_file(
-        path_or_fileobj=complete_html.encode(),
-        path_in_repo="index.html",
-        repo_id=space_name,
-        repo_type="space"
-    )
-    return f"https://huggingface.co/spaces/{space_name}"
+
+        # Upload HTML
+        upload_file(
+            path_or_fileobj=complete_html.encode(),
+            path_in_repo="index.html",
+            repo_id=space_name,
+            repo_type="space"
+        )
+
+        # Upload images if provided as bytes
+        for i, img in enumerate(images):
+            if isinstance(img, bytes):
+                upload_file(
+                    path_or_fileobj=img,
+                    path_in_repo=f"image_{i}.png",
+                    repo_id=space_name,
+                    repo_type="space"
+                )
+                images[i] = f"image_{i}.png"
+
+        logger.info(f"Updated space: {space_name}")
+        return f"https://huggingface.co/spaces/{space_name}"
+    except Exception as e:
+        logger.error(f"Error creating/updating space: {str(e)}")
+        return None
 
 def authenticate(username, password):
-    return username in AUTHORIZED_USERS and AUTHORIZED_USERS[username] == password
+    """Authenticate user against hardcoded credentials."""
+    if not username or not password:
+        logger.warning("Empty username or password provided")
+        return False
+    is_valid = username in AUTHORIZED_USERS and AUTHORIZED_USERS[username] == password
+    logger.info(f"Authentication attempt for {username}: {'Success' if is_valid else 'Failed'}")
+    return is_valid
 
 def generate_and_edit(topic, username, password):
+    """Generate blog and create editable space."""
+    logger.info(f"Starting generate_and_edit for topic: {topic}")
+    
+    # Check authentication
     if not authenticate(username, password):
-        return "Authentication failed", "", ""
+        return "Authentication failed: Incorrect username or password", "", "Error"
     
+    # Check HF_TOKEN
+    if not HF_TOKEN:
+        return "Authentication failed: HF_TOKEN not set in .env", "", "Error"
+
+    # Generate content
     initial_content = generate_initial_content(topic)
-    space_url = create_or_update_space(initial_content)
-    return space_url, initial_content, "Blog generated successfully"
+    if "Error" in initial_content:
+        return "Failed to generate content", initial_content, "Error"
+
+    # Generate images
+    image_prompts = [
+        f"Professional illustration about {topic}, clean design, minimalist style.",
+        f"Data visualization related to {topic}, infographic style."
+    ]
+    images = []
+    for prompt in image_prompts:
+        img_data = generate_image(prompt)
+        if img_data:
+            images.append(img_data)
+        time.sleep(2)  # Prevent rate limiting
 
-# Mock update endpoint (in practice, this would be a server-side API)
-def update_content(space, content, images):
-    create_or_update_space(content, space, images)
-    return "Updated successfully"
+    # Create space
+    space_url = create_or_update_space(initial_content, images=images)
+    if not space_url:
+        return "Failed to create space", initial_content, "Error"
+
+    return space_url, initial_content, "Blog generated successfully"
 
 # Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
@@ -691,7 +779,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     with gr.Row():
         with gr.Column():
             username = gr.Textbox(label="Username", placeholder="admin")
-            password = gr.Textbox(label="Password", type="password")
+            password = gr.Textbox(label="Password", type="password", placeholder="password123")
             topic_input = gr.Textbox(label="Blog Topic", placeholder="e.g., Future of AI")
             generate_btn = gr.Button("Generate & Edit", variant="primary")
         
@@ -700,9 +788,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             blog_link = gr.Markdown("Blog link will appear here...")
             blog_preview = gr.Markdown(label="Preview", value="Content preview will appear here...")
     
-    # Update function (simulated)
-    update_btn = gr.Button("Update Content", visible=False)
-    
     generate_btn.click(
         fn=generate_and_edit,
         inputs=[topic_input, username, password],
@@ -710,4 +795,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     )
 
 if __name__ == "__main__":
-    app.launch(share=True)
+    app.launch(share=True, debug=True)
+