Update app.py

app.py

@@ -6,7 +6,7 @@ from reportlab.lib.pagesizes import letter
 import tempfile
 from textwrap import wrap
 
-# Load API key securely from Hugging Face secrets
+# Load API key securely from environment
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=GROQ_API_KEY)
 
@@ -15,7 +15,6 @@ MOODS = [
     "Relaxing",
     "Romantic",
     "Cultural & Heritage",
-    "Nature & Wildlife",
     "Foodie & Culinary",
     "Shopping & Urban",
     "Beach & Island",
@@ -23,15 +22,14 @@ MOODS = [
 ]
 
 MOOD_IMAGES = {
-    "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"
+    "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.png"
 }
 
 def generate_itinerary(mood, location, budget, days):
@@ -59,7 +57,7 @@ def generate_itinerary(mood, location, budget, days):
     
     itinerary_text = completion.choices[0].message.content
     image_path = MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"])
-    return itinerary_text, image_path
+    return itinerary_text, image_path, gr.update(visible=True)
 
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
@@ -70,25 +68,24 @@ def generate_pdf(itinerary_text):
     width, height = letter
 
     margin = 40
-    max_width = width - 2 * margin
-    max_height = height - 2 * margin
-    line_height = 14
+    line_height = 12
+    font_size = 10
+    c.setFont("Courier", font_size)
+
+    max_chars_per_line = int((width - 2 * margin) / (font_size * 0.6))
+
     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
+        wrapped = wrap(paragraph, width=max_chars_per_line)
         lines.extend(wrapped if wrapped else [''])
 
     for line in lines:
         if y < margin + line_height:
             c.showPage()
-            c.setFont("Courier", 11)
+            c.setFont("Courier", font_size)
             y = height - margin
         c.drawString(x, y, line)
         y -= line_height
@@ -97,34 +94,33 @@ def generate_pdf(itinerary_text):
     return temp_pdf.name
 
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo:
-    # Header with animated "Feel Away" and italicized tagline in quotes
+    # Header with gr.HTML for styles and animation
+    header_html = """
+    <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 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>
+    """
     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 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>
-            """
-        )
+        gr.HTML(header_html)
 
     # Inputs
     with gr.Row():
@@ -135,24 +131,20 @@ 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")
     
-    # 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")
+        image_output = gr.Image(label="Travel Mood Image", type="filepath", visible=False)
 
-    # PDF download row
-    with gr.Row():
+    with gr.Column():
         download_btn = gr.Button("📥 Download Itinerary as PDF", variant="secondary")
         pdf_file = gr.File(label="Download PDF", file_types=[".pdf"])
 
-    # Event bindings
     generate_btn.click(
         generate_itinerary,
         inputs=[mood, location, budget, days],
-        outputs=[itinerary_output, image_output]
+        outputs=[itinerary_output, image_output, image_output]
     )
 
     download_btn.click(
@@ -162,4 +154,3 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")
     )
 
 demo.launch()
-