Update application file

app.py

@@ -275,7 +275,7 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
             raise ValueError("Generated plan is empty")
             
         # Format the plan using the new formatting function
-        formatted_plan = format_travel_plan(travel_plan, destination, days)
+        formatted_plan = format_travel_plan(travel_plan, destination, days, budget, interests)
         
         return formatted_plan
         
@@ -284,10 +284,10 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
         print(error_msg)  # Log the error
         
         # Generate a basic fallback plan
-        fallback_plan = generate_fallback_plan(destination, days, interests, budget)
+        fallback_plan = generate_fallback_plan(destination, days, budget, interests)
         return fallback_plan
     
-def generate_fallback_plan(destination, days, interests, budget):
+def generate_fallback_plan(destination, days, budget, interests):
     """
     Generate a basic fallback travel plan if the model fails
     
@@ -318,8 +318,8 @@ def generate_fallback_plan(destination, days, interests, budget):
         f"# Travel Plan Details",
         f"## Destination: {destination}",
         f"## Duration: {days} days",
-        f"## Interests: {', '.join(interests)}",
         f"## Budget: {budget}\n",
+        f"## Interests: {', '.join(interests)}",
         "## Itinerary:\n"
     ]
     
@@ -350,42 +350,111 @@ def generate_fallback_plan(destination, days, interests, budget):
     # Create a travel plan string
     return "\n".join(travel_plan_lines)
 
-def format_travel_plan(travel_plan, destination, days, interests=None, budget=None):
+def format_travel_plan(travel_plan, destination, days, interests, budget):
     """
-    Format travel plan string
+    Comprehensively format travel plan with enhanced customization
     
     Parameters:
     - travel_plan: Generated travel plan string
     - destination: Target destination
     - days: Number of trip days
-    - interests: Optional interests (default: None)
-    - budget: Optional budget (default: None)
+    - interests: List of user's interests
+    - budget: Budget level for the trip
     
     Returns:
-    Formatted travel plan as a string
+    Formatted and customized travel plan
     """
-    # If the input is already a properly formatted string, return it
-    if isinstance(travel_plan, str):
-        return travel_plan
+    # Validate inputs
+    if not destination or not isinstance(destination, str):
+        raise ValueError("Invalid destination")
+    
+    if not interests or not isinstance(interests, list):
+        raise ValueError("Interests must be a non-empty list")
+    
+    if not budget or budget not in ["Budget", "Moderate", "Luxury"]:
+        budget = "Moderate"  # Default to moderate if invalid
+    
+    # Ensure days is within reasonable range
+    days = max(1, min(14, int(days)))
+    
+    # Prepare the header with key trip details
+    formatted_lines = [
+        f"# 🌍 Travel Plan: {destination}",
+        f"## Trip Overview",
+        f"- **Destination:** {destination}",
+        f"- **Duration:** {days} {'day' if days == 1 else 'days'}",
+        f"- **Budget Level:** {budget}",
+        f"- **Interests:** {', '.join(interests)}",
+        "\n## Detailed Itinerary\n"
+    ]
     
-    # If it's a DataFrame or Series, extract the travel plan
+    # Try to parse the original travel plan
     try:
-        # Attempt to get the travel plan from DataFrame or Series
+        # If it's a DataFrame or Series, extract travel plan
         if hasattr(travel_plan, 'iloc'):
-            # If it's a DataFrame or Series, get the first row's travel plan
             travel_plan = travel_plan.iloc[0]['travel_plan']
         elif hasattr(travel_plan, 'get'):
-            # If it's a dictionary-like object
             travel_plan = travel_plan.get('travel_plan', '')
     except Exception:
-        # Fallback to generating a new plan if extraction fails
-        return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
+        travel_plan = ""
+    
+    # If no travel plan or empty, return a simple message
+    if not travel_plan or not isinstance(travel_plan, str):
+        return "\n".join(formatted_lines + [
+            "### 🚧 Plan Generation Unavailable",
+            "",
+            "Unfortunately, we couldn't generate a detailed travel plan at this moment. ",
+            "This could be due to temporary model limitations or processing issues.",
+            "",
+            "**Recommendations:**",
+            "- Try regenerating the plan",
+            "- Check your internet connection",
+            "- Verify the destination and interests",
+            "- Contact support if the issue persists"
+        ])
+    
+    # Combine header and travel plan
+    formatted_lines.append(travel_plan)
+    
+    # Final formatting and return
+    return "\n".join(formatted_lines)
+
+# def format_travel_plan(travel_plan, destination, days, interests, budget):
+#     """
+#     Format travel plan string
+    
+#     Parameters:
+#     - travel_plan: Generated travel plan string
+#     - destination: Target destination
+#     - days: Number of trip days
+#     - interests: Optional interests (default: None)
+#     - budget: Optional budget (default: None)
+    
+#     Returns:
+#     Formatted travel plan as a string
+#     """
+#     # If the input is already a properly formatted string, return it
+#     if isinstance(travel_plan, str):
+#         return travel_plan
+    
+#     # If it's a DataFrame or Series, extract the travel plan
+#     try:
+#         # Attempt to get the travel plan from DataFrame or Series
+#         if hasattr(travel_plan, 'iloc'):
+#             # If it's a DataFrame or Series, get the first row's travel plan
+#             travel_plan = travel_plan.iloc[0]['travel_plan']
+#         elif hasattr(travel_plan, 'get'):
+#             # If it's a dictionary-like object
+#             travel_plan = travel_plan.get('travel_plan', '')
+#     except Exception:
+#         # Fallback to generating a new plan if extraction fails
+#         return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
     
-    # Ensure travel_plan is a string
-    if not isinstance(travel_plan, str):
-        return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
+#     # Ensure travel_plan is a string
+#     if not isinstance(travel_plan, str):
+#         return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
     
-    return travel_plan
+#     return travel_plan
 
 
 def main():