Update app.py

app.py

@@ -1129,29 +1129,31 @@ def fetch_google_hotels(query="Birmingham Hotel", check_in="2024-08-14", check_o
 
 
 def fetch_google_flights(departure_id="JFK", arrival_id="BHM", outbound_date="2024-08-14", return_date="2024-08-20"):
+    from serpapi import GoogleSearch
+
     params = {
-        "engine": "google_flights",
-        "departure_id": departure_id,
-        "arrival_id": arrival_id,
-        "outbound_date": outbound_date,
-        "return_date": return_date,
-        "currency": "USD",
-        "hl": "en",
-        "api_key": os.getenv("SERP_API")
+      "engine": "google_flights",
+      "departure_id": "PEK",
+      "arrival_id": "AUS",
+      "outbound_date": "2024-08-14",
+      "return_date": "2024-08-20",
+      "currency": "USD",
+      "hl": "en",
+      "api_key": os.getenv("SERP_API")  # Replace with your actual API key
     }
     
     search = GoogleSearch(params)
     results = search.get_dict()
     
-    flight_info = ""
+    # Extract flight details from the results
     best_flights = results.get('best_flights', [])
+    formatted_flights = []
     
-    for flight in best_flights[:5]:  # Limiting to top 5 flights
-        flight_segments = flight.get('flights', [])
-        if not flight_segments:
-            continue
+    # Process each flight in the best_flights list
+    for i, flight in enumerate(best_flights, start=1):
+        flight_info = f"Flight {i}:\n"
         
-        for segment in flight_segments:
+        for segment in flight.get('flights', []):
             departure_airport = segment.get('departure_airport', {}).get('name', 'Unknown Departure Airport')
             departure_time = segment.get('departure_airport', {}).get('time', 'Unknown Time')
             arrival_airport = segment.get('arrival_airport', {}).get('name', 'Unknown Arrival Airport')
@@ -1159,15 +1161,21 @@ def fetch_google_flights(departure_id="JFK", arrival_id="BHM", outbound_date="20
             duration = segment.get('duration', 'Unknown Duration')
             airplane = segment.get('airplane', 'Unknown Airplane')
             
-            flight_info += f"**Flight Segment:**\n"
-            flight_info += f"  - **Departure:** {departure_airport} at {departure_time}\n"
-            flight_info += f"  - **Arrival:** {arrival_airport} at {arrival_time}\n"
-            flight_info += f"  - **Duration:** {duration} minutes\n"
-            flight_info += f"  - **Airplane:** {airplane}\n"
+            # Format the flight segment details
+            flight_info += f"  - Departure: {departure_airport} at {departure_time}\n"
+            flight_info += f"  - Arrival: {arrival_airport} at {arrival_time}\n"
+            flight_info += f"  - Duration: {duration} minutes\n"
+            flight_info += f"  - Airplane: {airplane}\n"
         
-        flight_info += "-" * 50 + "\n"
+        formatted_flights.append(flight_info)
+    
+    # Combine the formatted flight information into a single output
+    final_output = "Here are some available flights for today:\n\n" + "\n".join(formatted_flights)
+    
+   
+
 
-    return flight_info
+    return final_output