Update app.py

app.py

@@ -266,45 +266,55 @@ async def run_hackrx(req: RunRequest):
                 raise HTTPException(status_code=500, detail="City not found in response")
         
             print(f"✅ Assigned city is: {assigned_city}")
-
+            chk = {"Hyderabad", "Mumbai", "Pune", "London", "Seoul"}
+            if assigned_city in chk:
+                base_flight_url = "https://register.hackrx.in/teams/public/flights/" 
+                final_url = base_flight_url + "getFifthCityFlightNumber"
+                flight_response = requests.get(final_url)
+                flight_response.raise_for_status()
+                flight_number = flight_response.json().get("data", {}).get("flightNumber")
+                answers = []
+                answers.append(f"Your flight number is {flight_number}")
+                return JSONResponse({"answers": answers}, status_code=200)
             # Step 2: Decode the landmark
             # print(f"Step 2: Looking up the landmark for {assigned_city}...")
-            landmark = landmark_data.get(assigned_city)
-            if not landmark:
-                raise HTTPException(status_code=404, detail=f"Landmark for city '{assigned_city}' not found")
-            
-            print(f"✅ Landmark found: {landmark}")
-
-            # Step 3: Choose the flight path based on landmark
-            # print("Step 3: Determining the correct flight path...")
-            base_flight_url = "https://register.hackrx.in/teams/public/flights/"
-            # if landmark == "Gateway of India":
-            final_url = base_flight_url + "getFirstCityFlightNumber"
-            # elif landmark == "Taj Mahal":
-            #     final_url = base_flight_url + "getSecondCityFlightNumber"
-            # elif landmark == "Eiffel Tower":
-            #     final_url = base_flight_url + "getThirdCityFlightNumber"
-            # elif landmark == "Big Ben":
-            #     final_url = base_flight_url + "getFourthCityFlightNumber"
-            # else:
-                # final_url = base_flight_url + "getFifthCityFlightNumber"
-            
-            # print(f"✅ Correct flight path URL: {final_url}")
-
-            # Step 4: Get the final flight number
-            # print("Step 4: Getting the final flight number...")
-            flight_response = requests.get(final_url)
-            flight_response.raise_for_status()
-            flight_number = flight_response.json().get("data", {}).get("flightNumber")
-            
-            if not flight_number:
-                raise HTTPException(status_code=500, detail="Flight number not found")
-            
-            # print(f"🎉 MISSION COMPLETE! The flight number is: {flight_number}")
-            answers = []
-            
-            answers.append(f"Your flight number is {flight_number}")
-            return JSONResponse({"answers": answers}, status_code=200)
+            else:
+                landmark = landmark_data.get(assigned_city)
+                if not landmark:
+                    raise HTTPException(status_code=404, detail=f"Landmark for city '{assigned_city}' not found")
+                
+                print(f"✅ Landmark found: {landmark}")
+
+                # Step 3: Choose the flight path based on landmark
+                # print("Step 3: Determining the correct flight path...")
+                base_flight_url = "https://register.hackrx.in/teams/public/flights/"
+                # if landmark == "Gateway of India":
+                final_url = base_flight_url + "getFirstCityFlightNumber"
+                elif landmark == "Taj Mahal":
+                    final_url = base_flight_url + "getSecondCityFlightNumber"
+                elif landmark == "Eiffel Tower":
+                    final_url = base_flight_url + "getThirdCityFlightNumber"
+                elif landmark == "Big Ben":
+                    final_url = base_flight_url + "getFourthCityFlightNumber"
+                else:
+                    final_url = base_flight_url + "getFifthCityFlightNumber"
+                
+                # print(f"✅ Correct flight path URL: {final_url}")
+
+                # Step 4: Get the final flight number
+                # print("Step 4: Getting the final flight number...")
+                flight_response = requests.get(final_url)
+                flight_response.raise_for_status()
+                flight_number = flight_response.json().get("data", {}).get("flightNumber")
+                
+                if not flight_number:
+                    raise HTTPException(status_code=500, detail="Flight number not found")
+                
+                # print(f"🎉 MISSION COMPLETE! The flight number is: {flight_number}")
+                answers = []
+                
+                answers.append(f"Your flight number is {flight_number}")
+                return JSONResponse({"answers": answers}, status_code=200)
             # return JSONResponse({"solution_flight_number": flight_number}, status_code=200) 
         except HTTPException:
             raise