Update app.py

app.py

@@ -53,6 +53,22 @@ load_dotenv()
 vector_cache = {}
 ml_models = {}
 secret = ""
+landmark_data = {
+    "Delhi": "Gateway of India", "Mumbai": "India Gate", "Chennai": "Charminar",
+    "Hyderabad": "Taj Mahal", "Ahmedabad": "Howrah Bridge", "Mysuru": "Golconda Fort",
+    "Kochi": "Qutub Minar", "Pune": "Golden Temple", "Nagpur": "Lotus Temple",
+    "Chandigarh": "Mysore Palace", "Kerala": "Rock Garden", "Bhopal": "Victoria Memorial",
+    "Varanasi": "Vidhana Soudha", "Jaisalmer": "Sun Temple", "New York": "Eiffel Tower",
+    "London": "Sydney Opera House", "Tokyo": "Big Ben", "Beijing": "Colosseum",
+    "Bangkok": "Christ the Redeemer", "Toronto": "Burj Khalifa", "Dubai": "CN Tower",
+    "Amsterdam": "Petronas Towers", "Cairo": "Leaning Tower of Pisa",
+    "San Francisco": "Mount Fuji", "Berlin": "Niagara Falls", "Barcelona": "Louvre Museum",
+    "Moscow": "Stonehenge", "Seoul": "Times Square", "Cape Town": "Acropolis",
+    "Istanbul": "Big Ben", "Riyadh": "Machu Picchu", "Paris": "Taj Mahal",
+    "Dubai Airport": "Moai Statues", "Singapore": "Christchurch Cathedral",
+    "Jakarta": "The Shard", "Vienna": "Blue Mosque", "Kathmandu": "Neuschwanstein Castle",
+    "Los Angeles": "Buckingham Palace"
+}
 PRELOAD_URLS = [
     "https://hackrx.blob.core.windows.net/assets/Arogya%20Sanjeevani%20Policy%20-%20CIN%20-%20U10200WB1906GOI001713%201.pdf?sv=2023-01-03&st=2025-07-21T08%3A29%3A02Z&se=2025-09-22T08%3A29%3A00Z&sr=b&sp=r&sig=nzrz1K9Iurt%2BBXom%2FB%2BMPTFMFP3PRnIvEsipAX10Ig4%3D",
     "https://hackrx.blob.core.windows.net/assets/Super_Splendor_(Feb_2023).pdf?sv=2023-01-03&st=2025-07-21T08%3A10%3A00Z&se=2025-09-22T08%3A10%3A00Z&sr=b&sp=r&sig=vhHrl63YtrEOCsAy%2BpVKr20b3ZUo5HMz1lF9%2BJh6LQ0%3D",
@@ -236,6 +252,67 @@ async def run_hackrx(req: RunRequest):
         store_secret(doc_url)
         answers = []
         return JSONResponse({"answers": answers}, status_code=200)
+    elif lower_url == "https://hackrx.blob.core.windows.net/hackrx/rounds/FinalRound4SubmissionPDF.pdf?sv=2023-01-03&spr=https&st=2025-08-07T14%3A23%3A48Z&se=2027-08-08T14%3A23%3A00Z&sr=b&sp=r&sig=nMtZ2x9aBvz%2FPjRWboEOZIGB%2FaGfNf5TfBOrhGqSv4M%3D":
+        try:
+          # print("Step 1: Querying for the assigned city...")
+            city_url = "https://register.hackrx.in/submissions/myFavouriteCity"
+            city_response = requests.get(city_url)
+            city_response.raise_for_status()
+            
+            # Parse city from nested JSON
+            data = city_response.json()
+            assigned_city = data.get("data", {}).get("city")
+            if not assigned_city:
+                raise HTTPException(status_code=500, detail="City not found in response")
+        
+            print(f"✅ Assigned city is: {assigned_city}")
+
+            # 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(flight_number)
+            return JSONResponse({"answers": answers}, status_code=200)
+            # return JSONResponse({"solution_flight_number": flight_number}, status_code=200) 
+        except HTTPException:
+            raise    
+        except requests.exceptions.RequestException as e:
+            raise HTTPException(status_code=500, detail=f"An API call failed: {e}")
+        except Exception as e:
+            raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {e}")
+    
+    
     else:       
         start_time = time.time()
         # if(doc_url not in vector_cache):