Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import json, re, os, requests
+import csv
+from io import StringIO
 
-# المفاتيح من إعدادات الـ Space
+# API Keys from Space settings
 hf_token = os.getenv("HF_TOKEN")
 google_key = os.getenv("GOOGLE_MAPS_API_KEY")
 
-# إنشاء العميل
+# Initialize Client
 client = InferenceClient("Qwen/Qwen2.5-7B-Instruct", token=hf_token)
 
 SYSTEM_PROMPT = """Analyze the Arabic map query and return ONLY a JSON object with these exact keys:
@@ -31,7 +33,15 @@ SYSTEM_PROMPT = """Analyze the Arabic map query and return ONLY a JSON object wi
 
 Examples:
 Input: "أرخص وأقرب مستشفى طوارئ في جدة"
-Output: {"location": "جدة", "category": "مستشفى", "sub_type": "طوارئ", "features": ["أرخص", "أقرب"], "sort_by": "price"}"""
+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):
     """بناء جملة بحث محسّنة"""
@@ -64,7 +74,7 @@ def search_google_maps_new_api(parsed_data):
     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"
@@ -143,7 +153,7 @@ def search_google_maps_new_api(parsed_data):
             types = place.get("types", [])
             place_type = ", ".join([t.replace("_", " ").replace("restaurant", "مطعم").replace("cafe", "مقهى") 
                                    .replace("hospital", "مستشفى").replace("park", "حديقة") 
-                                   .replace("store", "متجر").replace("gas_station", "محطة وقود")[:15] 
+                                   .replace("store", "متجر").replace("gas_station", "محطة وقود") 
                                    for t in types[:2]])
             
             card = f"""### {i}. 📍 {name}
@@ -163,6 +173,9 @@ def search_google_maps_new_api(parsed_data):
 
 def search_google_maps_legacy(parsed_data, query=None):
     """الرجوع إلى Legacy API كاحتياط"""
+    if not google_key:
+        return "⚠️ الرجاء إضافة GOOGLE_MAPS_API_KEY في إعدادات الـ Space."
+
     if not query:
         query = build_search_query(parsed_data)
     
@@ -178,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", [])
         
@@ -253,11 +266,66 @@ def main_process(user_query):
     if "error" in parsed_data:
         return parsed_data, f"⚠️ {parsed_data.get('error', 'خطأ غير معروف')}"
     
-    # استخدام الـ New API مع fallback للـ Legacy
     map_results = search_google_maps_new_api(parsed_data)
     
     return parsed_data, map_results
 
+def process_batch_queries(file_obj, progress=gr.Progress()):
+    if file_obj is None:
+        return None, "❌ الرجاء تحميل ملف للاستعلامات الجماعية."
+
+    queries = []
+    try:
+        with open(file_obj.name, 'r', encoding='utf-8') as f:
+            for line in f:
+                query = line.strip()
+                if query:
+                    queries.append(query)
+    except Exception as e:
+        return None, f"❌ خطأ في قراءة الملف: {str(e)}"
+
+    if not queries:
+        return None, "❌ الملف فارغ أو لا يحتوي على استعلامات صالحة."
+
+    results = []
+    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)
+            if match:
+                top_result_name = match.group(1).strip()
+        elif "لم يتم العثور على نتائج" in map_results:
+            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', '  ') # 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())
+
+    return output_filepath, "✅ تم الانتهاء من معالجة الدفعة. يمكنك تحميل النتائج."
+
 # ==================== واجهة المستخدم ====================
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -268,41 +336,58 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     ⚠️ **تأكد من تفعيل Places API (New) في Google Cloud Console**
     """)
     
-    with gr.Row():
-        input_text = gr.Textbox(
-            label="📝 ماذا تبحث عنه؟",
-            placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض",
-            lines=2
+    with gr.Tab("استعلام فردي"):
+        with gr.Row():
+            input_text = gr.Textbox(
+                label="📝 ماذا تبحث عنه؟",
+                placeholder="مثال: أرخص مستشفى طوارئ في جدة، أو حديقة أطفال نظيفة في الرياض، أو فندق 5 نجوم في دبي",
+                lines=2
+            )
+        
+        btn = gr.Button("🔍 ابحث الآن", variant="primary")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                output_json = gr.JSON(label="⚙️ تحليل الذكاء الاصطناعي")
+            with gr.Column(scale=2):
+                output_map = gr.Markdown(label="🗺️ نتائج Google Maps")
+        
+        gr.Markdown("### 💡 أمثلة للبحث:")
+        examples = [
+            "أرخص وأقرب مستشفى طوارئ في جدة",
+            "حديقة أطفال نظيفة في الرياض",
+            "أفضل فندق 5 نجوم في دبي",
+            "مقهى هادئ للعمل في الخبر",
+            "محطة وقود 24 ساعة رخيصة",
+            "صيدلية قريبة مفتوحة الآن"
+        ]
+        
+        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)
+        
+        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)
+        
+        btn.click(fn=main_process, inputs=input_text, outputs=[output_json, output_map])
+
+    with gr.Tab("معالجة دفعة"):
+        gr.Markdown("""
+        ### ⬆️ تحميل ملف الاستعلامات
+        قم بتحميل ملف نصي (.txt) أو CSV يحتوي على استعلام واحد في كل سطر.
+        """)
+        file_input = gr.File(label="ملف الاستعلامات (TXT/CSV)", file_count="single", type="filepath")
+        batch_btn = gr.Button("🚀 ابدأ معالجة الدفعة", variant="primary")
+        batch_output_file = gr.File(label="ملف نتائج الدفعة (CSV)")
+        batch_status_message = gr.Markdown(label="الحالة")
+
+        batch_btn.click(
+            fn=process_batch_queries,
+            inputs=file_input,
+            outputs=[batch_output_file, batch_status_message]
         )
-    
-    btn = gr.Button("🔍 ابحث الآن", variant="primary")
-    
-    with gr.Row():
-        with gr.Column(scale=1):
-            output_json = gr.JSON(label="⚙️ تحليل الذكاء الاصطناعي")
-        with gr.Column(scale=2):
-            output_map = gr.Markdown(label="🗺️ نتائج Google Maps")
-    
-    gr.Markdown("### 💡 أمثلة للبحث:")
-    examples = [
-        "أرخص وأقرب مستشفى طوارئ في جدة",
-        "حديقة أطفال نظيفة في الرياض",
-        "أفضل فندق 5 نجوم في دبي",
-        "مقهى هادئ للعمل في الخبر",
-        "محطة وقود 24 ساعة رخيصة",
-        "صيدلية قريبة مفتوحة الآن"
-    ]
-    
-    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)
-    
-    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)
-    
-    btn.click(fn=main_process, inputs=input_text, outputs=[output_json, output_map])
 
-demo.launch()
+demo.launch()