Update services/location_service.py

services/location_service.py

@@ -4,6 +4,13 @@ from geopy.geocoders import Nominatim
 from geopy.exc import GeocoderTimedOut, GeocoderUnavailable
 from models.location_models import LocationData, Coordinates, ErrorResponse
 import requests
+from dotenv import load_dotenv
+load_dotenv()
+import json
+
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-1.5-flash')
 class LocationService:
     @staticmethod
     def extract_location_entities(ner_results):
@@ -23,7 +30,17 @@ class LocationService:
             return {k: v for k, v in {"city": city, "state": state, "country": country}.items() if v is not None}
         else:
             return None
-
+    @staticmethod
+    def system_prompt(location : str) -> str:
+        return f"""You are a Text Analyser where you will extract city , state , country from given piece of text given below.You will strictly extract following keys from the text country , state , city.
+        {location} \n
+        Rules:
+        1. You will analyse the text and extract the country , city or state from the text , lets say if you have 'Udhampur, JK, India' , here JK means Jammu and Kashmir , so if you get any initials extract the exact name. 
+        2. If any value is not found, return null.
+        3. If all values are null, return null.
+        Ensure the strictly that output is a valid JSON object containing strictly the above keys, without any explanations.
+        Generate a JSON response in the following format without using the ```json block. Ensure the output is properly formatted as plain text JSON.
+        """
     
     @staticmethod
     def get_coordinates(data:dict,model) -> Coordinates | ErrorResponse:
@@ -40,10 +57,10 @@ class LocationService:
         
         if location:
             # Assuming `app.nlp` is already initialized elsewhere and accessible
-            ner_results = model(location)
-            
+            llm_prompt = LocationService.system_prompt(location)
+            response = model.generate_content(llm_prompt)
             # Extract city, state, and country using the logic from extract_location_entities
-            location_entities = LocationService.extract_location_entities(ner_results)
+            location_entities = json.loads(response)
             
             if location_entities:
                 city = location_entities.get('city')