Update app.py

app.py

@@ -1,239 +1,232 @@
-from flask import Flask, render_template, request, send_file, redirect, url_for
-from PIL import Image, ImageDraw
-import tempfile
-import os
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+import datetime
 import requests
-import base64
-from io import BytesIO
+import pytz
+import yaml
+import re
+from tools.final_answer import FinalAnswerTool
+from Gradio_UI import GradioUI
 
-app = Flask(__name__)
-
-def generate_passover_coloring_page(age):
-    """
-    Generates a Passover-themed coloring page based on age.
+# Below is an example of a tool that does nothing. Amaze us with your creativity!
+@tool
+def create_new_vday_card_tool(theme: str) -> str:
+    # it's import to specify the return type
+    # Keep this format for the description / args / args description but feel free to modify the tool
+    """A tool that generates a custom Valentine's Day card based on a given theme.
+    Args:
+        theme: A string representing the theme for the Valentine's Day card.
     """
     try:
-        age = int(float(age))
-        if not (1 <= age <= 12):
-            return "Please enter a valid age between 1 and 12.", None
-    except (ValueError, TypeError):
-        return "Please enter a valid number for age.", None
-    
-    # Define complexity based on age
-    if 1 <= age <= 3:
-        complexity = "simple"
-        theme = "Matzah"
-        description = "Simple outline of matzah bread for coloring"
-    elif 4 <= age <= 6:
-        complexity = "basic"
-        theme = "Seder Plate"
-        description = "Basic outline of a Seder plate with clear sections"
-    elif 7 <= age <= 9:
-        complexity = "intermediate"
-        theme = "Exodus Story"
-        description = "Intermediate level scene from the Exodus story"
-    else:
-        complexity = "advanced"
-        theme = "Passover Symbols"
-        description = "Advanced design with various Passover symbols"
-    
-    # Create a coloring page based on the theme and complexity
-    img = Image.new('RGB', (600, 500), color='white')
-    d = ImageDraw.Draw(img)
-    
-    # Draw a border
-    d.rectangle([(20, 20), (580, 480)], outline='black', width=3)
-    
-    # Add title and information
-    d.text((50, 40), f"Passover Coloring Page", fill='black')
-    d.text((50, 70), f"Theme: {theme}", fill='black')
-    d.text((50, 100), f"For age: {age} (Complexity: {complexity})", fill='black')
-    d.text((50, 130), f"{description}", fill='black')
-    
-    # Draw content based on the theme
-    if theme == "Matzah":
-        # Matzah pattern
-        d.rectangle([(150, 180), (450, 380)], outline='black', width=2)
-        for i in range(160, 440, 25):
-            for j in range(190, 370, 25):
-                d.ellipse([(i, j), (i+5, j+5)], fill='black')
-                
-    elif theme == "Seder Plate":
-        # Seder plate
-        d.ellipse([(150, 180), (450, 380)], outline='black', width=2)
-        # Divide into sections
-        d.line([(300, 180), (300, 380)], fill='black', width=2)
-        d.line([(150, 280), (450, 280)], fill='black', width=2)
-        d.ellipse([(275, 255), (325, 305)], outline='black', width=2)
+        # Import tool from Hub
+        '''
+        image_generation_tool = Tool.from_space(
+            space_id="black-forest-labs/FLUX.1-schnell",
+            name="generate-image",
+            description="Generate image from prompt"
+        )
+        '''
+        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+        image = image_generation_tool(theme)
+        return final_answer(image)
+    except Exception as e:
+        return f"Error creating card for theme '{theme}': {str(e)}"
+
+# image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+final_answer = FinalAnswerTool()
+
+# Using Mistral-7B-Instruct-v0.1 instead of Qwen2.5-Coder-32B-Instruct
+model = HfApiModel(
+    max_tokens=2096,
+    temperature=0.7,  # Slightly higher temperature for Mistral as it's smaller
+    model_id='mistralai/Mistral-7B-Instruct-v0.1',  # Changed to Mistral model
+    # custom_role_conversions can be adjusted for Mistral
+    custom_role_conversions={
+        "system": "<s>[INST] <<SYS>>\n{message}\n<</SYS>>\n\n",
+        "human": "{message} [/INST]",
+        "assistant": "{message} </s><s>[INST] "
+    },
+)
+
+# Let's add a fix for the Gradio issue by patching the launch method
+def modified_launch():
+    try:
+        # Original code
+        with open("prompts.yaml", 'r') as stream:
+            prompt_templates = yaml.safe_load(stream)
         
-    elif theme == "Exodus Story":
-        # Simple boat/basket shape for baby Moses
-        d.arc([(150, 250), (450, 400)], 0, 180, fill='black', width=2)
-        d.line([(150, 325), (450, 325)], fill='black', width=2)
-        # Water waves
-        for i in range(170, 430, 50):
-            d.arc([(i, 350), (i+30, 370)], 0, 180, fill='black', width=2)
+        agent = CodeAgent(
+            model=model,
+            tools=[final_answer, create_new_vday_card_tool],  # add your tools here (don't remove final answer)
+            max_steps=6,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=prompt_templates
+        )
         
-    else:  # Passover Symbols
-        # Draw Star of David
-        center_x, center_y = 300, 280
-        radius = 80
+        # Get the Gradio UI instance but don't launch yet
+        ui = GradioUI(agent)
         
-        # First triangle
-        points1 = [
-            (center_x, center_y - radius),
-            (center_x + radius * 0.866, center_y + radius * 0.5),
-            (center_x - radius * 0.866, center_y + radius * 0.5),
-            (center_x, center_y - radius)
-        ]
+        # Patch the launch method to avoid API issues
+        original_launch = ui.launch
         
-        # Second triangle
-        points2 = [
-            (center_x, center_y + radius),
-            (center_x + radius * 0.866, center_y - radius * 0.5),
-            (center_x - radius * 0.866, center_y - radius * 0.5),
-            (center_x, center_y + radius)
-        ]
+        def patched_launch(*args, **kwargs):
+            # Try to patch the problematic API processing in routes.py
+            try:
+                import types
+                from gradio import routes
+                
+                # Create a dummy method that returns an empty dict to avoid the error
+                def dummy_api_info(self):
+                    return {"endpoints": {}, "components": [], "dependencies": []}
+                
+                # Try to patch FastAPI.api_info
+                try:
+                    if hasattr(routes, 'FastAPI'):
+                        routes.FastAPI.api_info = types.MethodType(dummy_api_info, routes.FastAPI)
+                except:
+                    pass
+                    
+            except:
+                pass
+                
+            # Call the original launch method
+            return original_launch(*args, **kwargs)
         
-        d.line(points1, fill='black', width=2)
-        d.line(points2, fill='black', width=2)
-    
-    # Convert image to base64 for HTML display
-    buffered = BytesIO()
-    img.save(buffered, format="PNG")
-    img_str = base64.b64encode(buffered.getvalue()).decode()
-    
-    # Create the response text
-    response = {
-        "age": age,
-        "theme": theme,
-        "description": description,
-        "complexity": complexity,
-        "image": img_str
-    }
-    
-    return response
-
-# Home page
-@app.route('/', methods=['GET', 'POST'])
-def home():
-    result = None
-    
-    if request.method == 'POST':
-        try:
-            age = request.form.get('age')
-            if age:
-                result = generate_passover_coloring_page(age)
-        except Exception as e:
-            return render_template('index.html', error=str(e))
-    
-    return render_template('index.html', result=result)
-
-# Create temporary templates directory and files if they don't exist
-def create_templates():
-    templates_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'templates')
-    if not os.path.exists(templates_dir):
-        os.makedirs(templates_dir)
+        # Replace the launch method with our patched version
+        ui.launch = patched_launch
+        
+        # Now launch the UI with our patched method
+        return ui.launch()
     
-    # Create index.html template
-    index_html = """
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <title>Passover Coloring Pages Generator</title>
-        <style>
-            body {
-                font-family: Arial, sans-serif;
-                max-width: 800px;
-                margin: 0 auto;
-                padding: 20px;
-            }
-            h1 {
-                color: #2c3e50;
-            }
-            .form-container {
-                margin: 20px 0;
-                padding: 15px;
-                border: 1px solid #ddd;
-                border-radius: 5px;
-            }
-            label {
-                display: block;
-                margin-bottom: 5px;
-                font-weight: bold;
-            }
-            input, button {
-                padding: 8px;
-                margin-bottom: 15px;
-            }
-            button {
-                background-color: #3498db;
-                color: white;
-                border: none;
-                cursor: pointer;
-                border-radius: 4px;
-            }
-            button:hover {
-                background-color: #2980b9;
-            }
-            .result-container {
-                margin-top: 20px;
-                padding: 15px;
-                border: 1px solid #ddd;
-                border-radius: 5px;
-            }
-            .image-container {
-                margin-top: 20px;
-                text-align: center;
-            }
-            .image-container img {
-                max-width: 100%;
-                border: 1px solid #ddd;
-            }
-            .error {
-                color: red;
-                font-weight: bold;
-            }
-        </style>
-    </head>
-    <body>
-        <h1>Passover Coloring Pages Generator</h1>
+    except Exception as e:
+        # If there's an error with Gradio, fall back to a Flask interface
+        print(f"Error with Gradio: {str(e)}")
+        print("Falling back to Flask interface...")
         
-        <div class="form-container">
-            <form method="POST">
-                <label for="age">Child's Age (1-12):</label>
-                <input type="number" id="age" name="age" min="1" max="12" value="6" required>
-                <button type="submit">Generate Coloring Page</button>
-            </form>
-        </div>
+        # Import Flask and create a simple interface
+        from flask import Flask, request, render_template_string, jsonify
+        import base64
+        from PIL import Image
+        from io import BytesIO
         
-        {% if error %}
-        <div class="error">
-            {{ error }}
-        </div>
-        {% endif %}
+        app = Flask(__name__)
         
-        {% if result %}
-        <div class="result-container">
-            <h2>Passover Coloring Page for a {{ result.age }}-year-old</h2>
-            <p><strong>Theme:</strong> {{ result.theme }}</p>
-            <p><strong>Description:</strong> {{ result.description }}</p>
-            <p><strong>Complexity:</strong> {{ result.complexity|capitalize }}</p>
+        # Simple HTML template
+        html_template = """
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <title>Valentine's Day Card Generator</title>
+            <style>
+                body {
+                    font-family: Arial, sans-serif;
+                    max-width: 800px;
+                    margin: 0 auto;
+                    padding: 20px;
+                }
+                h1 {
+                    color: #e91e63;
+                }
+                .form-container {
+                    margin: 20px 0;
+                    padding: 15px;
+                    border: 1px solid #ffcccc;
+                    border-radius: 5px;
+                    background-color: #fff0f5;
+                }
+                label {
+                    display: block;
+                    margin-bottom: 5px;
+                    font-weight: bold;
+                }
+                input {
+                    width: 100%;
+                    padding: 8px;
+                    margin-bottom: 15px;
+                    border: 1px solid #ccc;
+                    border-radius: 4px;
+                }
+                button {
+                    background-color: #e91e63;
+                    color: white;
+                    border: none;
+                    padding: 10px 15px;
+                    cursor: pointer;
+                    border-radius: 4px;
+                }
+                button:hover {
+                    background-color: #c2185b;
+                }
+                .result-container {
+                    margin-top: 20px;
+                    text-align: center;
+                }
+                .card-image {
+                    max-width: 100%;
+                    border: 10px solid #fff;
+                    box-shadow: 0 0 10px rgba(0,0,0,0.2);
+                }
+            </style>
+        </head>
+        <body>
+            <h1>Valentine's Day Card Generator</h1>
+            <p>Enter a theme to generate a custom Valentine's Day card.</p>
             
-            <div class="image-container">
-                <img src="data:image/png;base64,{{ result.image }}" alt="Passover Coloring Page">
+            <div class="form-container">
+                <form method="POST">
+                    <label for="theme">Card Theme:</label>
+                    <input type="text" id="theme" name="theme" placeholder="e.g., romantic sunset, cute animals, etc." required>
+                    <button type="submit">Generate Card</button>
+                </form>
             </div>
             
-            <p><em>Print this page for coloring!</em></p>
-        </div>
-        {% endif %}
-    </body>
-    </html>
-    """
-    
-    with open(os.path.join(templates_dir, 'index.html'), 'w') as f:
-        f.write(index_html)
+            {% if error %}
+            <div style="color: red; margin-top: 20px;">
+                <p>{{ error }}</p>
+            </div>
+            {% endif %}
+            
+            {% if image_data %}
+            <div class="result-container">
+                <h2>Your Valentine's Day Card</h2>
+                <p>Theme: {{ theme }}</p>
+                <img src="data:image/png;base64,{{ image_data }}" alt="Valentine's Day Card" class="card-image">
+            </div>
+            {% endif %}
+        </body>
+        </html>
+        """
+        
+        @app.route('/', methods=['GET', 'POST'])
+        def home():
+            if request.method == 'POST':
+                theme = request.form.get('theme', '')
+                if theme:
+                    try:
+                        # Add Valentine's context to the theme
+                        valentine_theme = f"Valentine's Day card with theme: {theme}, romantic, love, hearts"
+                        
+                        # Use the agent to generate the card
+                        image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+                        image = image_generation_tool(valentine_theme)
+                        
+                        # Convert PIL Image to base64 string
+                        buffered = BytesIO()
+                        image.save(buffered, format="PNG")
+                        img_str = base64.b64encode(buffered.getvalue()).decode()
+                        
+                        return render_template_string(html_template, image_data=img_str, theme=theme)
+                    except Exception as e:
+                        return render_template_string(html_template, error=f"Error: {str(e)}")
+                
+            return render_template_string(html_template)
+        
+        # Run the Flask app
+        return app.run(debug=True, host='0.0.0.0', port=8080)
 
-if __name__ == '__main__':
-    create_templates()
-    app.run(debug=True, host='0.0.0.0', port=8080)
+# Run the modified launch function
+if __name__ == "__main__":
+    modified_launch()