Update app.py

app.py

@@ -2,13 +2,16 @@ import os
 import gradio as gr
 import requests
 from openai import OpenAI
+import ast
 
-# API keys
+# API Keys
 weather_api_key = os.getenv("openweather")
 openai_api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=openai_api_key)
 
-# Fetch Weather
+# -------------------------------
+# Fetch Weather Data
+# -------------------------------
 def get_weather(city_name):
     if not city_name.strip():
         city_name = "Dubai"
@@ -16,7 +19,7 @@ def get_weather(city_name):
         url = f"https://api.openweathermap.org/data/2.5/weather?q={city_name}&appid={weather_api_key}&units=metric"
         data = requests.get(url).json()
 
-        if data["cod"] == 200:
+        if data.get("cod") == 200:
             rain = data.get("rain", {}).get("1h", 0)
             condition = data["weather"][0]["main"]
             emoji_map = {
@@ -38,13 +41,14 @@ def get_weather(city_name):
                 "visibility": data.get("visibility", 10000) // 1000,
                 "rain_chance": f"{rain} mm"
             }
-        else:
-            return None
-    except:
+        return None
+    except Exception:
         return None
 
 
-# Format Weather
+# -------------------------------
+# Weather Display HTML
+# -------------------------------
 def format_weather_display(data):
     if not data:
         return "<div style='text-align:center; color: #e74c3c; font-size: 18px; padding: 40px;'>❌ City not found. Please try again.</div>"
@@ -96,7 +100,9 @@ def format_weather_display(data):
     """
 
 
-# Chatbot using OpenAI
+# -------------------------------
+# Chatbot
+# -------------------------------
 def travel_chat(msg, history):
     messages = [{"role": "system", "content": "You are a helpful travel assistant. Suggest tourist attractions, activities, and travel tips for any city."}]
     for h in history:
@@ -118,7 +124,9 @@ def travel_chat(msg, history):
     return history, history
 
 
-# Fetch attractions for city
+# -------------------------------
+# Attractions
+# -------------------------------
 def get_attractions(city, country, temp, weather_desc):
     try:
         messages = [
@@ -139,19 +147,19 @@ def get_attractions(city, country, temp, weather_desc):
         )
 
         content = response.choices[0].message.content.strip()
-
         attractions = []
-        for line in content.split('\n'):
+        for line in content.splitlines():
             line = line.strip().rstrip(',')
-            if line.startswith('(') and line.endswith(')'):
+            if line.startswith("(") and line.endswith(")"):
                 try:
-                    attraction_tuple = eval(line)
+                    attraction_tuple = ast.literal_eval(line)
                     if isinstance(attraction_tuple, tuple) and len(attraction_tuple) == 2:
                         attractions.append(attraction_tuple)
-                except:
+                except Exception:
                     continue
+
         return attractions if attractions else [("No attractions found", "Try another city.")]
-    except:
+    except Exception:
         return [("Error", "Could not fetch attractions.")]
 
 
@@ -164,8 +172,9 @@ def format_attraction_card(name, details):
     """
 
 
+# -------------------------------
 # CSS
-# CSS
+# -------------------------------
 custom_css = """
 body, .gradio-container {
     background: linear-gradient(135deg, #f5f7fa 0%, #bbdefb 100%) !important;
@@ -176,7 +185,7 @@ body, .gradio-container {
     text-align: center;
     font-size: 2.8rem;
     font-weight: 800;
-    margin: 20px 0 15px 0;
+    margin: 20px 0;
     color: #0d47a1;
 }
 .section-header {
@@ -185,17 +194,10 @@ body, .gradio-container {
     padding: 12px 20px;
     border-radius: 12px 12px 0 0;
     margin: 0;
-    font-size: 1.2rem;
+    font-size: 1.3rem;
     font-weight: 600;
     text-align: center;
 }
-.section-subheading {
-    text-align: center;
-    font-size: 0.95rem;
-    color: #444;
-    margin: 10px 0 15px 0;
-    font-style: italic;
-}
 .content-box {
     background-color: #ffffff;
     border-radius: 0 0 16px 16px;
@@ -236,27 +238,43 @@ body, .gradio-container {
     text-align: center;
 }
 .footer {
-    background: linear-gradient(135deg, #0d47a1 0%, #002171 100%);
-    color: white;
-    padding: 20px;
     text-align: center;
-    margin-top: 40px;
     font-size: 14px;
-    border-radius: 12px;
+    padding: 20px;
+    color: #555;
+    margin-top: 20px;
 }
 """
 
+# -------------------------------
+# Generate Attraction Cards
+# -------------------------------
+def generate_attraction_cards(city):
+    weather = get_weather(city)
+    if not weather:
+        return "<div style='padding:20px; color:red;'>⚠️ Couldn't fetch attractions due to missing weather data.</div>"
+
+    attractions = get_attractions(
+        city=weather["city"],
+        country=weather["country"],
+        temp=weather["temperature"],
+        weather_desc=weather["description"]
+    )
+    return "<div class='card-grid'>" + "".join(format_attraction_card(name, details) for name, details in attractions) + "</div>"
+
+
+# -------------------------------
 # UI
+# -------------------------------
 def launch_ui():
-    with gr.Blocks(css=custom_css, title="Travel Weather Guide") as demo:
-        # HEADER
-        gr.Markdown("<div id='main-title'>🌍 Travel Weather Guide</div>")
+    with gr.Blocks(css=custom_css, title="TripMate AI") as demo:
+        # Main Title
+        gr.Markdown("<div id='main-title'>🌍 TripMate AI</div>")
 
         with gr.Row(equal_height=True):
             # Weather section
             with gr.Column(scale=1):
                 gr.Markdown("<div class='section-header'>🌤️ Weather Dashboard</div>")
-                gr.Markdown("<div class='section-subheading'>Check current weather details</div>")
                 with gr.Group(elem_classes="content-box"):
                     city_input = gr.Textbox(label="🏙️ City Name", value="Dubai")
                     update_btn = gr.Button("📍 Get Weather Data")
@@ -264,8 +282,7 @@ def launch_ui():
 
             # Chatbot section
             with gr.Column(scale=1):
-                gr.Markdown("<div class='section-header'>✈️ Travel Assistant</div>")
-                gr.Markdown("<div class='section-subheading'>Ask for travel tips and suggestions</div>")
+                gr.Markdown("<div class='section-header'>🤖 Travel Assistant</div>")
                 with gr.Group(elem_classes="content-box"):
                     chat = gr.Chatbot(height=350, type="messages")
                     with gr.Row():
@@ -275,16 +292,8 @@ def launch_ui():
 
         # Attractions section
         gr.Markdown("<div class='section-header'>🏖️ Top Attractions You Can Visit</div>")
-        gr.Markdown("<div class='section-subheading'>Plan your trip with these recommendations</div>")
         attractions_html = gr.HTML()
 
-        # FOOTER
-        gr.HTML("""
-        <div class='footer'>
-            Built with ❤️ | Powered by OpenWeather & OpenAI | © 2025 Travel Weather Guide
-        </div>
-        """)
-
         def update_all(city):
             return format_weather_display(get_weather(city)), generate_attraction_cards(city)
 
@@ -296,4 +305,10 @@ def launch_ui():
 
         demo.load(fn=lambda: update_all("Dubai"), inputs=None, outputs=[weather_html, attractions_html])
 
+        # Footer
+        gr.Markdown("<div class='footer'>© 2025 TripMate AI – Your AI-powered travel companion.</div>")
+
     demo.launch()
+
+
+launch_ui()