Upload app.py

app.py

@@ -5,6 +5,8 @@ import faiss
 import numpy as np
 from transformers import pipeline
 import time
+import ast
+import re
 
 # --- 1. DATA LOADING AND INITIALIZATION ---
 print("===== Application Startup =====")
@@ -34,6 +36,54 @@ except RuntimeError:
 
 
 # --- 3. SYNTHETIC GENERATION ---
+def format_plan_details(plan_string):
+    """
+    Parses and formats the raw plan string from the dataset into readable Markdown.
+    """
+    # If the plan is not in the expected dictionary format, return it as is.
+    if not plan_string or not plan_string.strip().startswith('['):
+        return plan_string
+
+    try:
+        # Safely parse the string representation of a list of dictionaries
+        plan_list = ast.literal_eval(plan_string)
+    except (ValueError, SyntaxError):
+        # If parsing fails, return the original string to avoid crashing
+        return plan_string
+
+    formatted_sections = []
+    for section in plan_list:
+        description = section.get('Description', 'Details')
+        content = section.get('Content', '').strip()
+        
+        # Add a bold title for each section
+        formatted_sections.append(f"#### {description}")
+
+        # Use specific formatting based on the section's description
+        if any(keyword in description for keyword in ['Attractions', 'Restaurants', 'Accommodations', 'Flight']):
+            lines = content.split('\n')
+            if lines:
+                # Make the header bold
+                formatted_sections.append(f"**{lines[0]}**")
+                # Format the rest of the lines as a clean, bulleted list
+                for item in lines[1:]:
+                    clean_item = ' '.join(item.split()) # Remove extra whitespace
+                    if clean_item:
+                        formatted_sections.append(f"- {clean_item}")
+        
+        elif 'Self-driving' in description or 'Taxi' in description:
+            # Make simple travel descriptions more readable
+            mode_emoji = "🚗" if 'Self-driving' in description else "🚕"
+            formatted_sections.append(f"- {mode_emoji} {content.replace(', ', ', ')}")
+        
+        else:
+            # Default formatting for any other type of content
+            formatted_sections.append(content)
+            
+        # Add a newline for spacing between sections
+        formatted_sections.append("") 
+
+    return "\n".join(formatted_sections)
 
 def get_recommendations_and_generate(query_text, k=3):
     # 1. Get Recommendations from existing data
@@ -87,9 +137,10 @@ def get_recommendations_and_generate(query_text, k=3):
 # --- 4. GRADIO USER INTERFACE ---
 def format_trip_plan(trip):
     # Formats the recommended trips with markdown
-    if not trip:
+    if not trip or 'reference_information' not in trip:
         return "### No similar trip plan found."
-    return f"### {trip['days']}-days trip to {trip['dest'].upper()}\n**Suggested Plan:**\n{trip['reference_information']}"
+    formatted_plan = format_plan_details(trip['reference_information'])
+    return f"### {trip['days']}-days trip to {trip['dest'].upper()}\n**Suggested Plan:**\n{formatted_plan}"
 
 def format_generated_trip(trip_text):
     return trip_text