Update app.py

app.py

@@ -8,17 +8,94 @@ 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. 
-Keys: location, category, cuisine, features (list), sort_by (rating, price, distance).
+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:
+   - Food: مطعم, مقهى, فرن, مخبز, سوق, بقالة, مطعم شاورما, مطعم برجر
+   - Shopping: مول, متجر, سوبرماركت, محل ملابس, صيدلية, محل إلكترونيات, محل أثاث
+   - Services: بنك, صراف آلي, محطة وقود, مغسلة, كوافير, ورشة سيارات, مكتب بريد
+   - Health: مستشفى, عيادة, صيدلية, طبيب أسنان, مركز صحي, طوارئ
+   - Education: مدرسة, جامعة, مكتبة, حضانة, معهد
+   - Entertainment: سينما, متحف, حديقة, ملعب, نادي رياضي, شاطئ, حديقة حيوان, منتزه
+   - Transportation: مطار, محطة قطار, محطة حافلات, موقف سيارات, تأجير سيارات, وقود
+   - Lodging: فندق, شقة فندقية, شاليه, مخيم, بيت ضيافة
+   - Religious: مسجد, كنيسة, معبد, جامع
+   - 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.
+
+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"}
 """
 
+# ==================== دوال البحث والمعالجة ====================
+
+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)
+    
+    return " ".join(parts)
+
 def search_google_maps(parsed_data):
     if not google_key:
-        return "الرجاء إضافة GOOGLE_MAPS_API_KEY في ا��إعدادات."
+        return "⚠️ الرجاء إضافة GOOGLE_MAPS_API_KEY في إعدادات الـ Space."
     
-    # بناء جملة البحث
-    query = f"{parsed_data.get('category', '')} {parsed_data.get('cuisine', '')} {parsed_data.get('location', '')} {' '.join(parsed_data.get('features', []))}"
+    # بناء جملة البحث المحسّنة
+    query = build_search_query(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 = {
@@ -27,52 +104,155 @@ def search_google_maps(parsed_data):
         "key": google_key
     }
     
-    response = requests.get(url, params=params)
-    results = response.json().get("results", [])
-    
-    # ترتيب النتائج إذا طلب المستخدم (اختياري)
-    if parsed_data.get("sort_by") == "rating":
-        results = sorted(results, key=lambda x: x.get("rating", 0), reverse=True)
-    
-    # تنسيق النتائج للعرض
-    output_list = []
-    for place in results[:5]: # عرض أول 5 نتائج
-        name = place.get("name")
-        rating = place.get("rating", "N/A")
-        address = place.get("formatted_address")
-        output_list.append(f"📍 **{name}**\n⭐ التقييم: {rating}\n🏠 العنوان: {address}\n---")
+    # إضافة معايير إضافية حسب نوع الترتيب
+    sort_by = parsed_data.get("sort_by", "relevance")
+    if sort_by == "distance":
+        # يمكن إضافة location bias لاحقاً إذا توفر GPS
+        pass
     
-    return "\n".join(output_list) if output_list else "لم يتم العثور على نتائج."
+    try:
+        response = requests.get(url, params=params, timeout=10)
+        data = response.json()
+        
+        if data.get("status") != "OK":
+            error_msg = data.get("error_message", data.get("status", "Unknown error"))
+            return f"⚠️ خطأ في API: {error_msg}"
+        
+        results = data.get("results", [])
+        
+        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 نتائج
+            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:
+                status = " ✅ مفتوح الآن"
+            elif open_now is False:
+                status = " ❌ مغلق الآن"
+            
+            # أنواع الأماكن (types)
+            types = place.get("types", [])
+            place_type = ", ".join([t.replace("_", " ") for t in types[:2]])
+            
+            card = f"""### {i}. 📍 {name}
+⭐ **التقييم:** {rating}/5 ({total_ratings} تقييم) {price_str}
+🏠 **العنوان:** {address}{status}
+🏷️ **النوع:** {place_type}
+---"""
+            output_list.append(card)
+        
+        # إضافة ملخص البحث
+        summary = f"🔍 **نتائج البحث عن:** `{query}` | **الترتيب حسب:** {sort_by}\n\n"
+        return summary + "\n".join(output_list)
+        
+    except requests.Timeout:
+        return "⏱️ انتهى وقت الانتظار. حاول مرة أخرى."
+    except Exception as e:
+        return f"❌ خطأ في الاتصال: {str(e)}"
 
 def main_process(user_query):
+    if not user_query.strip():
+        return {"error": "الرجاء إدخال استعلام"}, "لا توجد نتائج"
+    
     # 1. تحليل القصد باستخدام LLM
     try:
-        resp = client.chat.completions.create(
-            messages=[{"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": user_query}],
-            max_tokens=200, temperature=0.1
+        resp = client.chat_completions.create(
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": user_query}
+            ],
+            max_tokens=300,
+            temperature=0.1
         )
-        match = re.search(r'\{.*\}', resp.choices[0].message.content, re.DOTALL)
-        if not match: return "خطأ في تحليل البيانات", "لا توجد نتائج"
+        
+        raw_content = resp.choices[0].message.content
+        match = re.search(r'\{.*\}', raw_content, re.DOTALL)
+        
+        if not match:
+            return {
+                "error": "لم يتم العثور على JSON صالح",
+                "raw_response": raw_content
+            }, "خطأ في تحليل البيانات"
+        
         parsed_data = json.loads(match.group(0))
         
-        # 2. البحث في جوجل باستخدام البيانات المستخرجة
+        # التحقق من وجود الأخطاء في البيانات المستخرجة
+        if "error" in parsed_data:
+            return parsed_data, "فشل في تحليل الاستعلام"
+        
+        # 2. البحث في Google Maps
         map_results = search_google_maps(parsed_data)
         
         return parsed_data, map_results
+        
+    except json.JSONDecodeError as e:
+        return {"error": f"خطأ في قراءة JSON: {str(e)}"}, "فشل في تحليل البيانات"
     except Exception as e:
-        return {"error": str(e)}, "فشل في جلب البيانات"
+        return {"error": f"خطأ غير متوقع: {str(e)}"}, "فشل في جلب البيانات"
+
+# ==================== واجهة المستخدم ====================
 
-# واجهة المستخدم
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🗺️ تطبيق البحث الذكي في الخرائط")
+    gr.Markdown("""
+    # 🗺️ تطبيق البحث الذكي في الخرائط
+    **يدعم جميع الفئات:** مطاعم، فنادق، مستشفيات، محطات وقود، حدائق، متاجر، وغيرها...
+    """)
+    
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="📝 ماذا تبحث عنه؟",
+            placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض، أو فندق 5 نجوم في دبي",
+            lines=2
+        )
+    
+    btn = gr.Button("🔍 ابحث الآن", variant="primary", scale=1)
+    
     with gr.Row():
-        input_text = gr.Textbox(label="ماذا تبحث عنه؟", placeholder="مثال: مطعم شاورما في الرياض يكون نظيف وهادي")
+        with gr.Column(scale=1):
+            output_json = gr.JSON(label="⚙️ تحليل الذكاء الاصطناعي")
+        with gr.Column(scale=2):
+            output_map = gr.Markdown(label="🗺️ نتائج Google Maps")
     
-    btn = gr.Button("ابحث الآن 🔍", variant="primary")
+    # أمثلة سريعة
+    gr.Markdown("### 💡 أمثلة للبحث:")
+    examples = [
+        "أرخص وأقرب مستشفى طوارئ في جدة",
+        "حديقة أطفال نظيفة في الرياض",
+        "أفضل فندق 5 نجوم في دبي",
+        "مقهى هادئ للعمل في الخبر",
+        "محطة وقود 24 ساعة رخيصة",
+        "صيدلية قريبة مني مفتوحة الآن"
+    ]
+    
+    with gr.Row():
+        for i, ex in enumerate(examples[:3]):
+            gr.Button(ex, size="sm").click(lambda x=ex: x, None, input_text)
     
     with gr.Row():
-        output_json = gr.JSON(label="تحليل الذكاء الاصطناعي")
-        output_map = gr.Markdown(label="نتائج Google Maps")
+        for i, ex in enumerate(examples[3:]):
+            gr.Button(ex, size="sm").click(lambda x=ex: x, None, input_text)
     
     btn.click(fn=main_process, inputs=input_text, outputs=[output_json, output_map])