Update app.py

app.py

@@ -6,11 +6,11 @@ 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:
+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.
 2. "category": The main place type. Use standard Arabic terms:
@@ -54,8 +54,7 @@ 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": null, "features": ["هادئ", "للعمل"], "sort_by": "relevance"}"""
 
 # ==================== دوال البحث والمعالجة ====================
 
@@ -63,24 +62,19 @@ def build_search_query(parsed_data):
     """بناء جملة بحث محسّنة من البيانات المستخرجة"""
     parts = []
     
-    # نبدأ بالتصنيف الرئيسي
     category = parsed_data.get("category", "")
     if category:
         parts.append(category)
     
-    # نضيف النوع الفرعي إذا موجود
     sub_type = parsed_data.get("sub_type")
     if sub_type and sub_type != category:
         parts.append(str(sub_type))
     
-    # نضيف الموقع
     location = parsed_data.get("location", "")
     if location and location != "near me":
         parts.append(f"في {location}")
     
-    # نضيف الصفات المهمة للبحث (نختار الأكثر صلة)
     features = parsed_data.get("features", [])
-    # نصفات تساعد في البحث: 24 ساعة، طوارئ، أطفال، إلخ
     search_features = [f for f in features if any(keyword in f for keyword in 
         ["24", "طوارئ", "أطفال", "مجاني", "مفتوح", "فاخر", "اقتصادي"])]
     parts.extend(search_features)
@@ -91,7 +85,6 @@ def search_google_maps(parsed_data):
     if not google_key:
         return "⚠️ الرجاء إضافة GOOGLE_MAPS_API_KEY في إعدادات الـ Space."
     
-    # بناء جملة البحث المحسّنة
     query = build_search_query(parsed_data)
     
     if not query.strip():
@@ -104,11 +97,7 @@ def search_google_maps(parsed_data):
         "key": google_key
     }
     
-    # إضافة معايير إضافية حسب نوع الترتيب
     sort_by = parsed_data.get("sort_by", "relevance")
-    if sort_by == "distance":
-        # يمكن إضافة location bias لاحقاً إذا توفر GPS
-        pass
     
     try:
         response = requests.get(url, params=params, timeout=10)
@@ -123,28 +112,23 @@ def search_google_maps(parsed_data):
         if not results:
             return "❌ لم يتم العثور على نتائج. جرب تعديل البحث."
         
-        # ترتيب النتائج حسب المعيار المطلوب
         if sort_by == "rating":
             results = sorted(results, key=lambda x: x.get("rating", 0), reverse=True)
         elif sort_by == "price":
-            # Google Places API يعطي price_level (0-4)
             results = sorted(results, key=lambda x: x.get("price_level", 4))
         
-        # تنسيق النتائج للعرض
         output_list = []
-        for i, place in enumerate(results[:6], 1):  # عرض أول 6 نتائج
+        for i, place in enumerate(results[:6], 1):
             name = place.get("name", "غير معروف")
             rating = place.get("rating", "غير متوفر")
             total_ratings = place.get("user_ratings_total", 0)
             address = place.get("formatted_address", "العنوان غير متوفر")
             
-            # تحديد مستوى السعر
             price_level = place.get("price_level")
             price_str = ""
             if price_level:
                 price_str = "💰" * price_level
             
-            # حالة الافتتاح
             open_now = place.get("opening_hours", {}).get("open_now")
             status = ""
             if open_now is True:
@@ -152,7 +136,6 @@ def search_google_maps(parsed_data):
             elif open_now is False:
                 status = " ❌ مغلق الآن"
             
-            # أنواع الأماكن (types)
             types = place.get("types", [])
             place_type = ", ".join([t.replace("_", " ") for t in types[:2]])
             
@@ -163,7 +146,6 @@ def search_google_maps(parsed_data):
 ---"""
             output_list.append(card)
         
-        # إضافة ملخص البحث
         summary = f"🔍 **نتائج البحث عن:** `{query}` | **الترتيب حسب:** {sort_by}\n\n"
         return summary + "\n".join(output_list)
         
@@ -172,45 +154,61 @@ def search_google_maps(parsed_data):
     except Exception as e:
         return f"❌ خطأ في الاتصال: {str(e)}"
 
-def main_process(user_query):
-    if not user_query.strip():
-        return {"error": "الرجاء إدخال استعلام"}, "لا توجد نتائج"
-    
-    # 1. تحليل القصد باستخدام LLM
+def parse_arabic_query(user_query):
+    """تحليل الاستعلام باستخدام LLM - الطريقة الصحيحة"""
     try:
-        resp = client.chat_completions.create(
-            messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
-                {"role": "user", "content": user_query}
-            ],
+        # الطريقة الصحيحة لاستخدام 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,
-            temperature=0.1
+            temperature=0.1,
+            stream=False
         )
         
-        raw_content = resp.choices[0].message.content
-        match = re.search(r'\{.*\}', raw_content, re.DOTALL)
+        # استخراج المحتوى - قد يكون في أماكن مختلفة حسب الإصدار
+        if hasattr(response, 'choices') and response.choices:
+            raw_content = response.choices[0].message.content
+        elif isinstance(response, dict) and 'choices' in response:
+            raw_content = response['choices'][0]['message']['content']
+        else:
+            raw_content = str(response)
         
+        # البحث عن JSON في الرد
+        match = re.search(r'\{.*\}', raw_content, re.DOTALL)
         if not match:
             return {
                 "error": "لم يتم العثور على JSON صالح",
-                "raw_response": raw_content
-            }, "خطأ في تحليل البيانات"
+                "raw_response": raw_content[:500]
+            }
         
         parsed_data = json.loads(match.group(0))
-        
-        # التحقق من وجود الأخطاء في البيانات المستخرجة
-        if "error" in parsed_data:
-            return parsed_data, "فشل في تحليل الاستعلام"
-        
-        # 2. البحث في Google Maps
-        map_results = search_google_maps(parsed_data)
-        
-        return parsed_data, map_results
+        return parsed_data
         
     except json.JSONDecodeError as e:
-        return {"error": f"خطأ في قراءة JSON: {str(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)
+    
+    return parsed_data, map_results
 
 # ==================== واجهة المستخدم ====================
 
@@ -246,13 +244,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         "صيدلية قريبة مني مفتوحة الآن"
     ]
     
+    example_buttons = []
     with gr.Row():
-        for i, ex in enumerate(examples[:3]):
-            gr.Button(ex, size="sm").click(lambda x=ex: x, None, input_text)
+        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 i, ex in enumerate(examples[3:]):
-            gr.Button(ex, size="sm").click(lambda x=ex: x, None, input_text)
+        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])