Update app.py

app.py

@@ -13,6 +13,7 @@ from reportlab.lib.pagesizes import letter
 import spacy
 from geopy.geocoders import Nominatim
 import folium
+import time
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 
@@ -83,25 +84,43 @@ def generate_itinerary(mood, location, budget, days):
     budget_text = "any budget" if budget == "Any" else budget
     location_text = location if location.strip() != "" else "any destination"
     
+    # 🧠 AI prompt
     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"Create a detailed {days}-day travel itinerary for {location_text}. "
+        f"The traveler’s mood is {mood_text}, with a {budget_text} budget. "
+        f"Include daily activities, key landmarks, and meal or cultural suggestions. "
         f"Use Markdown formatting with **bold headings** for days and time slots."
     )
-    
+
+    # 🧩 Generate itinerary
     completion = client.chat.completions.create(
-        model="openai/gpt-oss-20b",
+        model="llama-3.1-8b",  # Use gemma or gpt-oss if this doesn't work
         messages=[
             {"role": "system", "content": "You are an expert travel planner."},
             {"role": "user", "content": prompt}
         ],
-        temperature=1,
-        max_completion_tokens=8192,
+        temperature=0.8,
+        max_tokens=2800,  # enough for multi-day itineraries
         top_p=1,
-        reasoning_effort="medium",
-        stream=False,
-        stop=None
+        stream=False
     )
+
+    itinerary_text = completion.choices[0].message.content
+
+    # 🌤️ Get general season/weather info
+    season_text = get_best_season(location_text)
+
+    # 📍 Map and image
+    image_path = os.path.abspath(MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"]))
+    map_html = extract_places_and_map(
+        itinerary_text, 
+        city_context=location_text if location_text.lower() != "any destination" else None
+    )
+
+    # 🧾 Combine output
+    combined_output = f"### {season_text}\n\n{itinerary_text}"
+    return combined_output, image_path, map_html
+
     
     itinerary_text = completion.choices[0].message.content
     image_path = os.path.abspath(MOOD_IMAGES.get(mood, MOOD_IMAGES["Any"]))
@@ -111,20 +130,20 @@ def generate_itinerary(mood, location, budget, days):
     return itinerary_text, image_path, map_html
 
 def get_best_season(location):
-    """AI-powered function to recommend the best season to visit any city, state, or country."""
+    """AI-powered function to recommend the best season or general weather for any location."""
     if not location.strip() or location.lower() == "any destination":
         return "🌤️ Best season info is available for specific locations only."
 
     prompt = (
-        f"Based on weather, tourism, and festivals, suggest the best season or months to visit {location}. "
-        f"Explain briefly in 25 words or less."
+        f"Suggest the best season or time of year to visit {location} based on weather and tourism. "
+        f"Keep it short and under 25 words."
     )
 
     try:
         response = client.chat.completions.create(
-            model="openai/gpt-oss-20b",  # You can replace this with any working model, e.g. "llama-3.1-8b"
+            model="llama-3.1-8b",  # or openai/gpt-oss-20b if supported
             messages=[
-                {"role": "system", "content": "You are an expert travel advisor."},
+                {"role": "system", "content": "You are a travel expert."},
                 {"role": "user", "content": prompt}
             ],
             temperature=0.7,
@@ -133,7 +152,7 @@ def get_best_season(location):
             stream=False
         )
 
-        # Safely extract the content depending on SDK format
+        # Extract content safely
         if hasattr(response.choices[0].message, "content"):
             season_text = response.choices[0].message.content
         else:
@@ -146,31 +165,34 @@ def get_best_season(location):
         return "🌤️ Unable to determine the best season right now."
 
 
+
 def generate_pdf(itinerary_text):
     if not itinerary_text.strip():
         return None, gr.update(visible=False)
 
+    # Unique file name for each download
     temp_dir = tempfile.gettempdir()
-    temp_path = os.path.join(temp_dir, "feelaway_itinerary.pdf")
-
-    if os.path.exists(temp_path):
-        os.remove(temp_path)
+    temp_filename = f"feelaway_itinerary_{int(time.time())}.pdf"
+    temp_path = os.path.join(temp_dir, temp_filename)
 
+    # Create a clean PDF
     doc = SimpleDocTemplate(temp_path, pagesize=letter,
                             rightMargin=40, leftMargin=40,
                             topMargin=40, bottomMargin=40)
 
-    # Register a Unicode font (DejaVu Sans is widely available)
-    pdfmetrics.registerFont(TTFont("DejaVuSans", "DejaVuSans.ttf"))
     styles = getSampleStyleSheet()
     style = styles["Normal"]
-    style.fontName = "DejaVuSans"  # Use Unicode-safe font
-    style.fontSize = 12
-    style.leading = 16
+    style.fontName = "Times-Roman"
+    style.fontSize = 13
+    style.leading = 18
 
+    # Convert markdown bold to HTML bold
     html_text = re.sub(r'\*\*(.+?)\*\*', r'<b>\1</b>', itinerary_text)
-    paragraphs = html_text.split('\n')
 
+    # Handle Markdown tables safely
+    html_text = re.sub(r'\|[^\n]*\|', lambda m: m.group(0).replace('|', '  '), html_text)
+
+    paragraphs = html_text.split('\n')
     story = []
     for para in paragraphs:
         p = Paragraph(para if para.strip() else ' ', style)
@@ -178,8 +200,8 @@ def generate_pdf(itinerary_text):
         story.append(Spacer(1, 6))
 
     doc.build(story)
-
     return temp_path, gr.update(visible=True)
+
 # ---------------------- UI ----------------------
 with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", secondary_hue="teal")) as demo: