Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 from crewai import Agent, Task, Crew, Process, LLM
-from crewai_tools import SerperDevTool, FileWriterTool
+from crewai_tools import FileWriterTool
 from dotenv import load_dotenv
 import traceback
 
@@ -11,7 +11,7 @@ load_dotenv()
 try:
     # Use CrewAI's LLM class with proper format for Gemini
     llm = LLM(
-        model="gemini/gemini-2.5-flash",
+        model="gemini/gemini-1.5-flash",
         temperature=0.6,
         api_key=os.environ.get("GOOGLE_API_KEY")
     )
@@ -23,9 +23,6 @@ except Exception as e:
     llm = None
 
 # --- 2. DEFINE TOOLS AND OUTPUT DIRECTORY ---
-# Using SerperDevTool for web search (you'll need SERPER_API_KEY in your environment)
-# Alternatively, you can remove the search tool if you don't have the API key
-search_tool = SerperDevTool()
 os.makedirs("outputs", exist_ok=True)
 file_write_tool = FileWriterTool()
 
@@ -34,19 +31,20 @@ designer = Agent(
   role='Principal Brand Designer',
   goal='Translate a user\'s "vibe" into a concrete, professional design brief for a website.',
   backstory=(
-      "You are a world-class brand designer with a keen eye for aesthetics. "
-      "You excel at understanding abstract concepts and turning them into actionable design elements."
+      "You are a world-class brand designer with a keen eye for aesthetics and deep knowledge of design trends. "
+      "You excel at understanding abstract concepts and turning them into actionable design elements. "
+      "You have extensive knowledge of color theory, typography, and modern web design best practices."
   ),
   verbose=True,
   llm=llm,
-  tools=[search_tool],
   allow_delegation=False
 )
 copywriter = Agent(
   role='Senior UX Copywriter',
   goal='Write compelling, vibe-appropriate copy for all sections of a website based on a design brief.',
   backstory=(
-      "You are an expert copywriter who specializes in creating website content that resonates with the target audience."
+      "You are an expert copywriter who specializes in creating website content that resonates with the target audience. "
+      "You understand how to match tone and style to brand identity."
   ),
   verbose=True,
   llm=llm,
@@ -56,7 +54,8 @@ developer = Agent(
   role='Senior Frontend Developer',
   goal='Take a design brief and content map to build a complete, self-contained HTML file.',
   backstory=(
-      "You are a skilled frontend developer who can build beautiful and functional websites from scratch."
+      "You are a skilled frontend developer who can build beautiful and functional websites from scratch. "
+      "You are expert in HTML5, CSS3, responsive design, and creating visually stunning web experiences."
   ),
   verbose=True,
   llm=llm,
@@ -68,33 +67,53 @@ developer = Agent(
 design_task = Task(
   description=(
       "Analyze the user's prompt: '{prompt}' to understand the desired vibe. "
-      "Research inspiration on color palettes, font pairings, and overall aesthetics. "
-      "Create a comprehensive 'Design Brief' document with a color palette (hex codes), font pairing (Google Fonts), and a detailed page structure."
+      "Using your extensive design knowledge, create a comprehensive 'Design Brief' document that includes:\n"
+      "1. A carefully selected color palette with 3-5 colors (provide hex codes)\n"
+      "2. Font pairing recommendations using Google Fonts (specify primary and secondary fonts)\n"
+      "3. Overall aesthetic direction (modern, minimalist, bold, etc.)\n"
+      "4. Detailed page structure with sections (hero, about, portfolio/services, contact, etc.)\n"
+      "5. Design principles to follow (spacing, imagery style, etc.)"
   ),
   expected_output=(
-      "A structured markdown document with the complete Design Brief (Color Palette, Font Pairing, Page Structure)."
+      "A structured markdown document with the complete Design Brief including Color Palette (with hex codes), "
+      "Font Pairing (Google Fonts names), Aesthetic Direction, Page Structure, and Design Principles."
   ),
   agent=designer
 )
 copywriting_task = Task(
   description=(
       "Using the Design Brief, write all text content for the website. "
-      "Ensure the tone matches the vibe. Create placeholder content for all sections."
+      "Ensure the tone and style match the specified vibe perfectly. "
+      "Create engaging, professional copy for each section including:\n"
+      "- Hero section headline and subheadline\n"
+      "- About section text\n"
+      "- Service/Portfolio descriptions\n"
+      "- Call-to-action text\n"
+      "- Contact section copy\n"
+      "Make the content authentic and compelling."
   ),
   expected_output=(
-      "A structured markdown 'Content Map' with text for each section of the website."
+      "A structured markdown 'Content Map' with polished, ready-to-use text for each section of the website."
   ),
   agent=copywriter,
   context=[design_task]
 )
 development_task = Task(
   description=(
-      "Based on the Design Brief and Content Map, create a single-file HTML website. "
-      "The file must be self-contained with embedded CSS and any simple JavaScript. "
-      "Use the specified colors and Google Fonts. Save the HTML to 'outputs/index.html' using the FileWriterTool."
+      "Based on the Design Brief and Content Map, create a complete, production-ready, single-file HTML website. "
+      "Requirements:\n"
+      "- Self-contained HTML file with embedded CSS (in <style> tags)\n"
+      "- Use the exact colors from the design brief (hex codes)\n"
+      "- Import and use the specified Google Fonts\n"
+      "- Fully responsive design that works on mobile, tablet, and desktop\n"
+      "- Modern, clean code with proper semantic HTML5\n"
+      "- Include smooth scrolling and subtle animations where appropriate\n"
+      "- Use the content from the Content Map\n"
+      "- Save the complete HTML to 'outputs/index.html' using the FileWriterTool\n\n"
+      "The file path for FileWriterTool should be: outputs/index.html"
   ),
   expected_output=(
-      "The complete HTML code for the website, written to 'outputs/index.html'."
+      "A complete, functional HTML website file saved to 'outputs/index.html'."
   ),
   agent=developer,
   context=[design_task, copywriting_task],