Update app.py

app.py

@@ -1,13 +1,47 @@
 import os
 import gradio as gr
 from groq import Groq
+from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import letter
+import tempfile
 
-# Load API key securely
+# Load API key securely from Hugging Face secrets
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
+MOODS = [
+    "Adventure",
+    "Relaxing",
+    "Romantic",
+    "Cultural & Heritage",
+    "Foodie & Culinary",
+    "Shopping & Urban",
+    "Beach & Island",
+    "Any"
+]
+
+MOOD_IMAGES = {
+    "Adventure": "images/adventure.png",
+    "Relaxing": "images/relaxing.png",
+    "Romantic": "images/romantic.png",
+    "Cultural & Heritage": "images/cultural.png",
+    "Foodie & Culinary": "images/foodie.png",
+    "Shopping & Urban": "images/shopping.png",
+    "Beach & Island": "images/beach.png",
+    "Any": "images/default.jpg"
+}
+
 def generate_itinerary(mood, location, budget, days):
-    prompt = f"Create a {days}-day travel itinerary to {location}. The mood of the traveler is {mood} and the budget is {budget}."
+    mood_text = "any mood" if mood == "Any" else mood
+    budget_text = "any budget" if budget == "Any" else budget
+    location_text = location if location.strip() != "" else "any destination"
+    
+    prompt = (
+        f"Create a {days}-day travel itinerary to {location_text}. "
+        f"The mood of the traveler is {mood_text} and the budget is {budget_text}. "
+        f"Use Markdown formatting with **bold headings** for days and time slots."
+    )
+    
     completion = client.chat.completions.create(
         model="gemma2-9b-it",
         messages=[
@@ -19,24 +53,75 @@ def generate_itinerary(mood, location, budget, days):
         top_p=1,
         stream=False
     )
-    return completion.choices[0].message.content
+    
+    itinerary_text = completion.choices[0].message.content
+    image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
+    return itinerary_text, image_path
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Mood-based Travel Itinerary Planner")
+def generate_pdf(itinerary_text):
+    if not itinerary_text.strip():
+        return None
+    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    c = canvas.Canvas(temp_pdf.name, pagesize=letter)
+    width, height = letter
+    text_obj = c.beginText(40, height - 40)
+    text_obj.setFont("Helvetica", 11)
+    
+    for line in itinerary_text.splitlines():
+        text_obj.textLine(line)
+    c.drawText(text_obj)
+    c.showPage()
+    c.save()
     
+    return temp_pdf.name
+
+with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
+    # Header
     with gr.Row():
-        mood = gr.Dropdown(
-            ["Adventure", "Relaxing", "Romantic", "Cultural"], 
-            label="Mood"
+        gr.Markdown(
+            """
+            <div style='text-align: center;'>
+                <h1 style='color: #006d6d; font-size: 38px; margin-bottom: 0;'>Feel Away</h1>
+                <p style='font-size: 18px; color: #007777; margin-top: 4px;'>
+                    Your Mood, Your Journey – Personalized Itineraries, Instantly
+                </p>
+            </div>
+            """
         )
-        location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France")
+
+    # Inputs
+    with gr.Row():
+        mood = gr.Dropdown(MOODS, label="Mood", value="Any", info="Select the mood of your trip")
+        location = gr.Textbox(label="Destination", placeholder="e.g., India, Japan, France (leave blank for Any)", info="Where do you want to go?")
     
     with gr.Row():
-        budget = gr.Dropdown(["Low", "Medium", "High"], label="Budget")
-        days = gr.Number(value=5, precision=0, label="Days")
+        budget = gr.Dropdown(["Low", "Medium", "High", "Any"], label="Budget", value="Any", info="Choose your budget level")
+        days = gr.Slider(1, 15, value=5, step=1, label="Days", info="Number of days for the itinerary")
     
-    output = gr.Textbox(label="Generated Itinerary", lines=20)
-    btn = gr.Button("Generate Itinerary")
-    btn.click(generate_itinerary, inputs=[mood, location, budget, days], outputs=output)
+    # Buttons
+    with gr.Row():
+        generate_btn = gr.Button("✨ Generate Itinerary", variant="primary")
+    
+    # Outputs
+    with gr.Row():
+        image_output = gr.Image(label="Travel Mood Image", type="filepath")
+        itinerary_output = gr.Markdown(label="Generated Itinerary")
+    
+    with gr.Row():
+        download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
+        pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
+    
+    # Events
+    generate_btn.click(
+        generate_itinerary, 
+        inputs=[mood, location, budget, days], 
+        outputs=[itinerary_output, image_output]
+    )
+    
+    download_btn.click(
+        generate_pdf, 
+        inputs=[itinerary_output], 
+        outputs=pdf_file
+    )
 
 demo.launch()