lamma

app.py

@@ -3,16 +3,13 @@ from huggingface_hub import InferenceClient
 import json, re, os, requests
 import csv
 from io import StringIO
-import functools
-import time
 
 # API Keys from Space settings
 hf_token = os.getenv("HF_TOKEN")
 google_key = os.getenv("GOOGLE_MAPS_API_KEY")
 
-# Initialize Client with ALLaM model
-# Note: Using the preview instruct version of ALLaM for best Arabic dialect support
-    client = InferenceClient("sdaia/allam-7b", token=hf_token)
+# Initialize Client
+client = InferenceClient("meta-llama/Meta-Llama-3.1-8B-Instruct", token=hf_token)
 
 SYSTEM_PROMPT = """Analyze the Arabic map query and return ONLY a JSON object with these exact keys:
 
@@ -69,18 +66,17 @@ def build_search_query(parsed_data):
     
     return " ".join(parts)
 
-@functools.lru_cache(maxsize=128)
-def search_google_maps_new_api(parsed_data_json):
+def search_google_maps_new_api(parsed_data):
     """استخدام Places API (New) v1"""
-    parsed_data = json.loads(parsed_data_json)
     if not google_key:
         return "⚠️ الرجاء إضافة GOOGLE_MAPS_API_KEY في إعدادات الـ Space."
     
     query = build_search_query(parsed_data)
     
     if not query.strip():
-        query = f"{parsed_data.get('category', '')} {parsed_data.get('location', '')}"
+        query = f"{parsed_data.get("category", "")} {parsed_data.get("location", "")}"
     
+    # Places API (New) - Text Search v1
     url = "https://places.googleapis.com/v1/places:searchText"
     
     headers = {
@@ -95,7 +91,11 @@ def search_google_maps_new_api(parsed_data_json):
         "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.post(url, headers=headers, json=data, timeout=15)
@@ -103,6 +103,7 @@ def search_google_maps_new_api(parsed_data_json):
         
         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}"
@@ -112,6 +113,7 @@ def search_google_maps_new_api(parsed_data_json):
         if not places:
             return "❌ لم يتم العثور على نتائج. جرب تعديل البحث."
         
+        # ترتيب النتائج
         if sort_by == "rating":
             places = sorted(places, key=lambda x: x.get("rating", 0), reverse=True)
         elif sort_by == "price":
@@ -120,6 +122,7 @@ def search_google_maps_new_api(parsed_data_json):
                           "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(places[:6], 1):
             name = place.get("displayName", {}).get("text", "غير معروف")
@@ -127,6 +130,7 @@ def search_google_maps_new_api(parsed_data_json):
             total_ratings = place.get("userRatingCount", 0)
             address = place.get("formattedAddress", "العنوان غير متوفر")
             
+            # مستوى السعر
             price_level = place.get("priceLevel")
             price_map = {
                 "PRICE_LEVEL_INEXPENSIVE": "💰",
@@ -136,10 +140,16 @@ def search_google_maps_new_api(parsed_data_json):
             }
             price_str = price_map.get(price_level, "")
             
+            # حالة الافتتاح
             hours = place.get("currentOpeningHours", {})
             open_now = hours.get("openNow")
-            status = " ✅ مفتوح الآن" if open_now is True else " ❌ مغلق الآن" if open_now is False else ""
+            status = ""
+            if open_now is True:
+                status = " ✅ مفتوح الآن"
+            elif open_now is False:
+                status = " ❌ مغلق الآن"
             
+            # الأنواع
             types = place.get("types", [])
             place_type = ", ".join([t.replace("_", " ").replace("restaurant", "مطعم").replace("cafe", "مقهى") 
                                    .replace("hospital", "مستشفى").replace("park", "حديقة") 
@@ -181,7 +191,7 @@ def search_google_maps_legacy(parsed_data, query=None):
         data = response.json()
         
         if data.get("status") != "OK":
-            return f"⚠️ خطأ في Legacy API أيضاً: {data.get('status')}. الرجاء تفعيل Places API (New) في Google Cloud Console."
+            return f"⚠️ خطأ في Legacy API أيضاً: {data.get("status")}. الرجاء تفعيل Places API (New) في Google Cloud Console."
         
         results = data.get("results", [])
         
@@ -215,9 +225,8 @@ def search_google_maps_legacy(parsed_data, query=None):
     except Exception as e:
         return f"❌ فشل في Legacy API أيضاً: {str(e)}"
 
-@functools.lru_cache(maxsize=128)
 def parse_arabic_query(user_query):
-    """تحليل الاستعلام باستخدام ALLaM LLM"""
+    """تحليل الاستعلام باستخدام LLM"""
     try:
         messages = [
             {"role": "system", "content": SYSTEM_PROMPT},
@@ -226,7 +235,7 @@ def parse_arabic_query(user_query):
         
         response = client.chat_completion(
             messages=messages,
-            max_tokens=150, # Optimized for speed
+            max_tokens=300,
             temperature=0.1,
             stream=False
         )
@@ -249,7 +258,6 @@ def parse_arabic_query(user_query):
         return {"error": f"خطأ في التحليل: {str(e)}"}
 
 def main_process(user_query):
-    start_time = time.perf_counter()
     if not user_query.strip():
         return {"error": "الرجاء إدخال استعلام"}, "لا توجد نتائج"
     
@@ -258,16 +266,9 @@ def main_process(user_query):
     if "error" in parsed_data:
         return parsed_data, f"⚠️ {parsed_data.get('error', 'خطأ غير معروف')}"
     
-    # Convert dict to string for caching
-    parsed_data_json = json.dumps(parsed_data, sort_keys=True)
-    map_results = search_google_maps_new_api(parsed_data_json)
+    map_results = search_google_maps_new_api(parsed_data)
     
-    end_time = time.perf_counter()
-    response_time_ms = (end_time - start_time) * 1000
-    
-    map_results_with_time = f"⏱️ وقت الاستجابة: {response_time_ms:.2f} مللي ثانية\n\n" + map_results
-    
-    return parsed_data, map_results_with_time
+    return parsed_data, map_results
 
 def process_batch_queries(file_obj, progress=gr.Progress()):
     if file_obj is None:
@@ -290,8 +291,10 @@ def process_batch_queries(file_obj, progress=gr.Progress()):
     for i, query in enumerate(progress.tqdm(queries, desc="معالجة الاستعلامات")):
         parsed_data, map_results = main_process(query)
         
+        # Flatten parsed_data for CSV, handle errors
         parsed_json_str = json.dumps(parsed_data, ensure_ascii=False) if "error" not in parsed_data else parsed_data.get("error", "")
         
+        # Extract top result name if available, otherwise indicate no results
         top_result_name = ""
         if "نتائج البحث عن" in map_results and "📍" in map_results:
             match = re.search(r'### \d+\. 📍 ([^\n]+)', map_results)
@@ -299,22 +302,24 @@ def process_batch_queries(file_obj, progress=gr.Progress()):
                 top_result_name = match.group(1).strip()
         elif "لم يتم العثور على نتائج" in map_results:
             top_result_name = "لا توجد نتائج"
-        else:
-            top_result_name = "خطأ أو لا توجد نتائج"
+        elif "خطأ" in map_results:
+            top_result_name = f"خطأ في API: {map_results}"
 
         results.append({
             "Original Query": query,
             "Parsed JSON": parsed_json_str,
             "Top Map Result": top_result_name,
-            "Full Map Results": map_results.replace('\n', '  ')
+            "Full Map Results": map_results.replace('\n', '  ') # Replace newlines for CSV readability
         })
 
+    # Write results to a CSV in memory
     output_buffer = StringIO()
     fieldnames = ["Original Query", "Parsed JSON", "Top Map Result", "Full Map Results"]
     writer = csv.DictWriter(output_buffer, fieldnames=fieldnames)
     writer.writeheader()
     writer.writerows(results)
     
+    # Save to a temporary file for Gradio to handle
     output_filepath = "./batch_results.csv"
     with open(output_filepath, 'w', encoding='utf-8') as f:
         f.write(output_buffer.getvalue())
@@ -325,17 +330,17 @@ def process_batch_queries(file_obj, progress=gr.Progress()):
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 🗺️ تطبيق البحث الذكي في الخرائط (مدعوم بـ ALLaM)
+    # 🗺️ تطبيق البحث الذكي في الخرائط
     **يدعم جميع الفئات:** مطاعم، فنادق، مستشفيات، محطات وقود، حدائق، متاجر، وغيرها...
     
-    🚀 **تم التحديث لاستخدام نموذج ALLaM المتخصص في اللغة العربية لنتائج أدق وأسرع.**
+    ⚠️ **تأكد من تفعيل Places API (New) في Google Cloud Console**
     """)
     
     with gr.Tab("استعلام فردي"):
         with gr.Row():
             input_text = gr.Textbox(
                 label="📝 ماذا تبحث عنه؟",
-                placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض",
+                placeholder="��ثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض، أو فندق 5 نجوم في دبي",
                 lines=2
             )
         
@@ -343,7 +348,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         
         with gr.Row():
             with gr.Column(scale=1):
-                output_json = gr.JSON(label="⚙️ تحليل ALLaM")
+                output_json = gr.JSON(label="⚙️ تحليل الذكاء الاصطناعي")
             with gr.Column(scale=2):
                 output_map = gr.Markdown(label="🗺️ نتائج Google Maps")