Update app.py

app.py

@@ -4,6 +4,7 @@ from groq import Groq
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
 import tempfile
+from textwrap import wrap
 
 # Load API key securely from Hugging Face secrets
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
@@ -14,6 +15,7 @@ MOODS = [
     "Relaxing",
     "Romantic",
     "Cultural & Heritage",
+    "Nature & Wildlife",
     "Foodie & Culinary",
     "Shopping & Urban",
     "Beach & Island",
@@ -21,13 +23,14 @@ MOODS = [
 ]
 
 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",
+    "Adventure": "images/adventure.jpg",
+    "Relaxing": "images/relaxing.jpg",
+    "Romantic": "images/romantic.jpg",
+    "Cultural & Heritage": "images/cultural.jpg",
+    "Nature & Wildlife": "images/nature.jpg",
+    "Foodie & Culinary": "images/foodie.jpg",
+    "Shopping & Urban": "images/shopping.jpg",
+    "Beach & Island": "images/beach.jpg",
     "Any": "images/default.jpg"
 }
 
@@ -61,30 +64,64 @@ def generate_itinerary(mood, location, budget, days):
 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()
+
+    margin = 40
+    max_width = width - 2 * margin
+    max_height = height - 2 * margin
+    line_height = 14
+    x = margin
+    y = height - margin
+
+    # Use a fixed-width font for better alignment
+    c.setFont("Courier", 11)
+
+    # Split text into paragraphs and wrap each to fit the page width
+    lines = []
+    for paragraph in itinerary_text.split('\n'):
+        wrapped = wrap(paragraph, width=95)  # adjust width for wrapping
+        lines.extend(wrapped if wrapped else [''])
+
+    for line in lines:
+        if y < margin + line_height:
+            c.showPage()
+            c.setFont("Courier", 11)
+            y = height - margin
+        c.drawString(x, y, line)
+        y -= line_height
+
     c.save()
-    
     return temp_pdf.name
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
-    # Header
+    # Header with animated "Feel Away" and italicized tagline in quotes
     with gr.Row():
         gr.Markdown(
             """
+            <style>
+            @keyframes colorCycle {
+                0% { color: #008080; }
+                25% { color: #20b2aa; }
+                50% { color: #40e0d0; }
+                75% { color: #008080; }
+                100% { color: #20b2aa; }
+            }
+            .color-cycle {
+                animation: colorCycle 4s infinite;
+            }
+            .tagline {
+                font-style: italic;
+                color: #007777;
+                font-size: 18px;
+                margin-top: 4px;
+            }
+            </style>
             <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>
+                <h1 class="color-cycle" style="font-size: 38px; margin-bottom: 0;">FEEL AWAY APP</h1>
+                <p class="tagline">"Your Mood, Your Journey – Personalized Itineraries, Instantly"</p>
             </div>
             """
         )
@@ -98,30 +135,31 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
         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")
     
-    # 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")
+    # Generate button
+    generate_btn = gr.Button("✨ Generate Itinerary", variant="primary")
+
+    # Outputs stacked vertically: itinerary markdown then image below
+    with gr.Column():
         itinerary_output = gr.Markdown(label="Generated Itinerary")
-    
+        image_output = gr.Image(label="Travel Mood Image", type="filepath")
+
+    # PDF download row
     with gr.Row():
         download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
         pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
-    
-    # Events
+
+    # Event bindings
     generate_btn.click(
-        generate_itinerary, 
-        inputs=[mood, location, budget, days], 
+        generate_itinerary,
+        inputs=[mood, location, budget, days],
         outputs=[itinerary_output, image_output]
     )
-    
+
     download_btn.click(
-        generate_pdf, 
-        inputs=[itinerary_output], 
+        generate_pdf,
+        inputs=[itinerary_output],
         outputs=pdf_file
     )
 
 demo.launch()
+