Update app.py

app.py

@@ -6,10 +6,9 @@ import json, re, os, requests
 hf_token = os.getenv("HF_TOKEN")
 google_key = os.getenv("GOOGLE_MAPS_API_KEY")
 
-# إنشاء العميل - الطريقة الصحيحة
+# إنشاء العميل
 client = InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=hf_token)
 
-# ==================== SYSTEM PROMPT المُحسَّن ====================
 SYSTEM_PROMPT = """Analyze the Arabic map query and return ONLY a JSON object with these exact keys:
 
 1. "location": City, area, or "near me" if not specified.
@@ -26,40 +25,16 @@ SYSTEM_PROMPT = """Analyze the Arabic map query and return ONLY a JSON object wi
    - Government: ديوان بلدية, محكمة, مركز شرطة, سفارة, مركز حكومي
    - Outdoor: شاطئ, ممر مشاة, بحيرة, غابة, جبل, منتزه طبيعي
 
-3. "sub_type": Specific type within category (e.g., "شاورما" for مطعم, "أطفال" for حديقة, "طوارئ" for مستشفى, "5 نجوم" for فندق) or null.
-
-4. "features": A list of ALL descriptive tags mentioned (e.g., ["رخيص", "قريب", "نظيف", "ممتاز", "24 ساعة", "مجاني", "هادئ", "عائلي", "فاخر", "سريع", "جيد"]).
-
-5. "sort_by": Standardized ranking signal. Use ONE of: rating, price, distance, relevance, open_now.
-
-Rules:
-- Extract ALL adjectives into "features", don't miss any.
-- If user mentions "أقرب/قريب/أقرب مني" → sort_by: "distance"
-- If user mentions "أحسن/أفضل/أعلى تقييم/ممتاز" → sort_by: "rating"
-- If user mentions "أرخص/رخيص/سعر منخفض" → sort_by: "price"
-- If user mentions "مفتوح الآن/مفتوح 24 ساعة" → sort_by: "open_now"
-- Default to "relevance" if unclear.
+3. "sub_type": Specific type within category or null.
+4. "features": A list of ALL descriptive tags mentioned.
+5. "sort_by": ONE of: rating, price, distance, relevance, open_now.
 
 Examples:
 Input: "أرخص وأقرب مستشفى طوارئ في جدة"
-Output: {"location": "جدة", "category": "مستشفى", "sub_type": "طوارئ", "features": ["أرخص", "أقرب"], "sort_by": "price"}
-
-Input: "حديقة أطفال نظيفة في الرياض"
-Output: {"location": "الرياض", "category": "حديقة", "sub_type": "أطفال", "features": ["نظيفة"], "sort_by": "relevance"}
-
-Input: "أفضل فندق 5 نجوم قريب من المطار في دبي"
-Output: {"location": "دبي", "category": "فندق", "sub_type": "5 نجوم", "features": ["أفضل", "قريب من المطار"], "sort_by": "rating"}
-
-Input: "محطة وقود 24 ساعة رخيصة"
-Output: {"location": "near me", "category": "محطة وقود", "sub_type": null, "features": ["24 ساعة", "رخيصة"], "sort_by": "price"}
-
-Input: "مقهى هادئ للعمل في الرياض"
-Output: {"location": "الرياض", "category": "مقهى", "sub_type": null, "features": ["هادئ", "للعمل"], "sort_by": "relevance"}"""
-
-# ==================== دوال البحث والمعالجة ====================
+Output: {"location": "جدة", "category": "مستشفى", "sub_type": "طوارئ", "features": ["أرخص", "أقرب"], "sort_by": "price"}"""
 
 def build_search_query(parsed_data):
-    """بناء جملة بحث محسّنة من البيانات المستخرجة"""
+    """بناء جملة بحث محسّنة"""
     parts = []
     
     category = parsed_data.get("category", "")
@@ -76,12 +51,13 @@ def build_search_query(parsed_data):
     
     features = parsed_data.get("features", [])
     search_features = [f for f in features if any(keyword in f for keyword in 
-        ["24", "طوارئ", "أطفال", "مجاني", "مفتوح", "فاخر", "اقتصادي"])]
+        ["24", "طوارئ", "أطفال", "مجاني", "مفتوح", "فاخر", "اقتصادي", "رخيص"])]
     parts.extend(search_features)
     
     return " ".join(parts)
 
-def search_google_maps(parsed_data):
+def search_google_maps_new_api(parsed_data):
+    """استخدام Places API (New) v1"""
     if not google_key:
         return "⚠️ الرجاء إضافة GOOGLE_MAPS_API_KEY في إعدادات الـ Space."
     
@@ -90,54 +66,85 @@ def search_google_maps(parsed_data):
     if not query.strip():
         query = f"{parsed_data.get('category', '')} {parsed_data.get('location', '')}"
     
-    url = "https://maps.googleapis.com/maps/api/place/textsearch/json"
-    params = {
-        "query": query,
-        "language": "ar",
-        "key": google_key
+    # Places API (New) - Text Search v1
+    url = "https://places.googleapis.com/v1/places:searchText"
+    
+    headers = {
+        "Content-Type": "application/json",
+        "X-Goog-Api-Key": google_key,
+        "X-Goog-FieldMask": "places.displayName,places.formattedAddress,places.rating,places.userRatingCount,places.priceLevel,places.currentOpeningHours,places.types,places.location"
     }
     
+    data = {
+        "textQuery": query,
+        "languageCode": "ar",
+        "maxResultCount": 10
+    }
+    
+    # إضافة تفضيل الموقع إذا كان الترتيب حسب المسافة
     sort_by = parsed_data.get("sort_by", "relevance")
+    if sort_by == "distance" and parsed_data.get("location") != "near me":
+        # يمكن إضافة locationBias لاحقاً
+        pass
     
     try:
-        response = requests.get(url, params=params, timeout=10)
-        data = response.json()
+        response = requests.post(url, headers=headers, json=data, timeout=15)
+        result = response.json()
         
-        if data.get("status") != "OK":
-            error_msg = data.get("error_message", data.get("status", "Unknown error"))
+        if response.status_code != 200:
+            error_msg = result.get("error", {}).get("message", "Unknown error")
+            # إذا كان الخطأ بسبب عدم تفعيل API، نحاول الـ Legacy كاحتياط
+            if "not enabled" in error_msg.lower() or response.status_code == 403:
+                return search_google_maps_legacy(parsed_data, query)
             return f"⚠️ خطأ في API: {error_msg}"
         
-        results = data.get("results", [])
+        places = result.get("places", [])
         
-        if not results:
+        if not places:
             return "❌ لم يتم العثور على نتائج. جرب تعديل البحث."
         
+        # ترتيب النتائج
         if sort_by == "rating":
-            results = sorted(results, key=lambda x: x.get("rating", 0), reverse=True)
+            places = sorted(places, key=lambda x: x.get("rating", 0), reverse=True)
         elif sort_by == "price":
-            results = sorted(results, key=lambda x: x.get("price_level", 4))
+            price_order = {"PRICE_LEVEL_UNSPECIFIED": 0, "PRICE_LEVEL_FREE": 1, 
+                          "PRICE_LEVEL_INEXPENSIVE": 2, "PRICE_LEVEL_MODERATE": 3,
+                          "PRICE_LEVEL_EXPENSIVE": 4, "PRICE_LEVEL_VERY_EXPENSIVE": 5}
+            places = sorted(places, key=lambda x: price_order.get(x.get("priceLevel"), 99))
         
+        # تنسيق النتائج
         output_list = []
-        for i, place in enumerate(results[:6], 1):
-            name = place.get("name", "غير معروف")
+        for i, place in enumerate(places[:6], 1):
+            name = place.get("displayName", {}).get("text", "غير معروف")
             rating = place.get("rating", "غير متوفر")
-            total_ratings = place.get("user_ratings_total", 0)
-            address = place.get("formatted_address", "العنوان غير متوفر")
+            total_ratings = place.get("userRatingCount", 0)
+            address = place.get("formattedAddress", "العنوان غير متوفر")
             
-            price_level = place.get("price_level")
-            price_str = ""
-            if price_level:
-                price_str = "💰" * price_level
+            # مستوى السعر
+            price_level = place.get("priceLevel")
+            price_map = {
+                "PRICE_LEVEL_INEXPENSIVE": "💰",
+                "PRICE_LEVEL_MODERATE": "💰💰",
+                "PRICE_LEVEL_EXPENSIVE": "💰💰💰",
+                "PRICE_LEVEL_VERY_EXPENSIVE": "💰💰💰💰"
+            }
+            price_str = price_map.get(price_level, "")
             
-            open_now = place.get("opening_hours", {}).get("open_now")
+            # حالة الافتتاح
+            hours = place.get("currentOpeningHours", {})
+            open_now = hours.get("openNow")
             status = ""
             if open_now is True:
                 status = " ✅ مفتوح الآن"
             elif open_now is False:
                 status = " ❌ مغلق الآن"
             
+            # الأنواع
             types = place.get("types", [])
-            place_type = ", ".join([t.replace("_", " ") for t in types[:2]])
+            place_type = ", ".join([t.replace("_", " ").replace("restaurant", "مطعم").replace("cafe", "مقهى") 
+                                   .replace("hospital", "مستشفى").replace("park", "حديقة") 
+                                   .replace("store", "متجر").replace("gas_station", "محطة وقود")[:15] 
+                                   for t in types[:2]])
             
             card = f"""### {i}. 📍 {name}
 ⭐ **التقييم:** {rating}/5 ({total_ratings} تقييم) {price_str}
@@ -154,16 +161,65 @@ def search_google_maps(parsed_data):
     except Exception as e:
         return f"❌ خطأ في الاتصال: {str(e)}"
 
+def search_google_maps_legacy(parsed_data, query=None):
+    """الرجوع إلى Legacy API كاحتياط"""
+    if not query:
+        query = build_search_query(parsed_data)
+    
+    url = "https://maps.googleapis.com/maps/api/place/textsearch/json"
+    params = {
+        "query": query,
+        "language": "ar",
+        "key": google_key
+    }
+    
+    try:
+        response = requests.get(url, params=params, timeout=10)
+        data = response.json()
+        
+        if data.get("status") != "OK":
+            return f"⚠️ خطأ في Legacy API أيضاً: {data.get('status')}. الرجاء تفعيل Places API (New) في Google Cloud Console."
+        
+        results = data.get("results", [])
+        
+        if not results:
+            return "❌ لم يتم العثور على نتائج."
+        
+        sort_by = parsed_data.get("sort_by", "relevance")
+        if sort_by == "rating":
+            results = sorted(results, key=lambda x: x.get("rating", 0), reverse=True)
+        
+        output_list = []
+        for i, place in enumerate(results[:6], 1):
+            name = place.get("name", "غير معروف")
+            rating = place.get("rating", "غير متوفر")
+            address = place.get("formatted_address", "العنوان غير متوفر")
+            
+            price_level = place.get("price_level")
+            price_str = "💰" * price_level if price_level else ""
+            
+            open_now = place.get("opening_hours", {}).get("open_now")
+            status = " ✅ مفتوح" if open_now else " ❌ مغلق" if open_now is False else ""
+            
+            card = f"""### {i}. 📍 {name}
+⭐ **التقييم:** {rating}/5 {price_str}
+🏠 **العنوان:** {address}{status}
+---"""
+            output_list.append(card)
+        
+        return "\n".join(output_list)
+        
+    except Exception as e:
+        return f"❌ فشل في Legacy API أيضاً: {str(e)}"
+
 def parse_arabic_query(user_query):
-    """تحليل الاستعلام باستخدام LLM - الطريقة الصحيحة"""
+    """تحليل الاستعلام باستخدام LLM"""
     try:
-        # الطريقة الصحيحة لاستخدام InferenceClient مع النماذج المحادثة
         messages = [
             {"role": "system", "content": SYSTEM_PROMPT},
             {"role": "user", "content": user_query}
         ]
         
-        # استخدام chat_completion (مع underscore) أو text_generation
         response = client.chat_completion(
             messages=messages,
             max_tokens=300,
@@ -171,7 +227,6 @@ def parse_arabic_query(user_query):
             stream=False
         )
         
-        # استخراج المحتوى - قد يكون في أماكن مختلفة حسب الإصدار
         if hasattr(response, 'choices') and response.choices:
             raw_content = response.choices[0].message.content
         elif isinstance(response, dict) and 'choices' in response:
@@ -179,34 +234,27 @@ def parse_arabic_query(user_query):
         else:
             raw_content = str(response)
         
-        # البحث عن JSON في الرد
         match = re.search(r'\{.*\}', raw_content, re.DOTALL)
         if not match:
-            return {
-                "error": "لم يتم العثور على JSON صالح",
-                "raw_response": raw_content[:500]
-            }
+            return {"error": "لم يتم العثور على JSON صالح", "raw_response": raw_content[:500]}
         
         parsed_data = json.loads(match.group(0))
         return parsed_data
         
-    except json.JSONDecodeError as e:
-        return {"error": f"خطأ في قراءة JSON: {str(e)}"}
     except Exception as e:
-        return {"error": f"خطأ في تحليل الاستعلام: {str(e)}"}
+        return {"error": f"خطأ في التحليل: {str(e)}"}
 
 def main_process(user_query):
     if not user_query.strip():
         return {"error": "الرجاء إدخال استعلام"}, "لا توجد نتائج"
     
-    # 1. تحليل القصد
     parsed_data = parse_arabic_query(user_query)
     
     if "error" in parsed_data:
         return parsed_data, f"⚠️ {parsed_data.get('error', 'خطأ غير معروف')}"
     
-    # 2. البحث في Google Maps
-    map_results = search_google_maps(parsed_data)
+    # استخدام الـ New API مع fallback للـ Legacy
+    map_results = search_google_maps_new_api(parsed_data)
     
     return parsed_data, map_results
 
@@ -216,16 +264,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
     # 🗺️ تطبيق البحث الذكي في الخرائط
     **يدعم جميع الفئات:** مطاعم، فنادق، مستشفيات، محطات وقود، حدائق، متاجر، وغيرها...
+    
+    ⚠️ **تأكد من تفعيل Places API (New) في Google Cloud Console**
     """)
     
     with gr.Row():
         input_text = gr.Textbox(
             label="📝 ماذا تبحث عنه؟",
-            placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض، أو فندق 5 نجوم في دبي",
+            placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض",
             lines=2
         )
     
-    btn = gr.Button("🔍 ابحث الآن", variant="primary", scale=1)
+    btn = gr.Button("🔍 ابحث ا��آن", variant="primary")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -233,7 +283,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=2):
             output_map = gr.Markdown(label="🗺️ نتائج Google Maps")
     
-    # أمثلة سريعة
     gr.Markdown("### 💡 أمثلة للبحث:")
     examples = [
         "أرخص وأقرب مستشفى طوارئ في جدة",
@@ -241,21 +290,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         "أفضل فندق 5 نجوم في دبي",
         "مقهى هادئ للعمل في الخبر",
         "محطة وقود 24 ساعة رخيصة",
-        "صيدلية قريبة مني مفتوحة الآن"
+        "صيدلية قريبة مفتوحة الآن"
     ]
     
-    example_buttons = []
     with gr.Row():
         for ex in examples[:3]:
             btn_ex = gr.Button(ex, size="sm")
             btn_ex.click(lambda x=ex: x, None, input_text)
-            example_buttons.append(btn_ex)
     
     with gr.Row():
         for ex in examples[3:]:
             btn_ex = gr.Button(ex, size="sm")
             btn_ex.click(lambda x=ex: x, None, input_text)
-            example_buttons.append(btn_ex)
     
     btn.click(fn=main_process, inputs=input_text, outputs=[output_json, output_map])