Update app.py

app.py

@@ -3,19 +3,18 @@ from tools.arabic_generator import ArabicTextGenerator
 from tools.quran_search import QuranSearchEngine
 from tools.tool_agent import ToolCallingAgent
 
-# Initialize all models
+# Initialize models
 text_gen = ArabicTextGenerator()
-quran_searcher = QuranSearchEngine()
-tool_agent = ToolCallingAgent()
+quran = QuranSearchEngine()
+tool_agent = ToolCallingAgent()  # CPU-only
 
-# Define tools for the agent
 TOOLS = [
     {
         "name": "search_quran",
-        "description": "Search the Quran for verses",
+        "description": "Search Quran verses",
         "parameters": {
             "query": {"type": "string"},
-            "max_results": {"type": "integer"}
+            "max_results": {"type": "integer", "max": 5}
         }
     },
     {
@@ -23,60 +22,50 @@ TOOLS = [
         "description": "Generate Arabic text",
         "parameters": {
             "prompt": {"type": "string"},
-            "length": {"type": "integer"}
+            "length": {"type": "integer", "max": 150}
         }
     }
 ]
 
-def execute_tool_call(tool_call):
-    """Execute the tool and return results"""
-    if tool_call["tool_name"] == "search_quran":
-        return quran_searcher.search(
-            tool_call["parameters"]["query"],
-            tool_call["parameters"]["max_results"]
-        )
-    elif tool_call["tool_name"] == "generate_text":
-        return text_gen.generate(
-            tool_call["parameters"]["prompt"],
-            tool_call["parameters"]["length"]
-        )
-    return "Tool not found"
+def format_output(result):
+    if isinstance(result, list):  # Quran results
+        return "\n".join(f"{r['surah']} {r['ayah']}: {r['text']}" for r in result)
+    return str(result)
 
-with gr.Blocks(title="Advanced Arabic Tools") as app:
-    # Tab 1: Arabic Text Generator
+with gr.Blocks(title="الأدوات العربية") as app:
+    # Tab 1: Existing Arabic Generator
     with gr.Tab("🖊️ مولد النصوص"):
-        text_input = gr.Textbox(label="النص الأولي")
-        length_slider = gr.Slider(50, 300, value=100, label="طول النص")
-        gen_btn = gr.Button("توليد")
-        text_output = gr.Textbox(label="النص المولد", lines=6)
-        
-        gen_btn.click(
-            text_gen.generate,
-            inputs=[text_input, length_slider],
-            outputs=text_output
-        )
+        # ... [Keep your existing Arabic generator UI exactly as is] ...
 
-    # Tab 2: Quran Search
+    # Tab 2: Existing Quran Search
     with gr.Tab("📖 القرآن الكريم"):
-        search_input = gr.Textbox(label="موضوع البحث")
-        top_k = gr.Slider(1, 10, value=5, label="عدد النتائج")
-        search_btn = gr.Button("ابحث")
-        search_output = gr.Textbox(label="النتائج", lines=10)
-        
-        search_btn.click(
-            lambda q, k: format_quran_results(quran_searcher.search(q, k)),
-            inputs=[search_input, top_k],
-            outputs=search_output
-        )
+        # ... [Keep your existing Quran search UI exactly as is] ...
 
-    # Tab 3: Tool Calling Agent
-    with gr.Tab("🛠️ الوكيل الذكي"):
-        tool_input = gr.Textbox(label="أدخل طلبك")
-        tool_output = gr.JSON(label="استجابة الأداة")
-        tool_btn = gr.Button("تنفيذ")
+    # Tab 3: New Tool Agent (CPU-optimized)
+    with gr.Tab("🛠️ مساعد ذكي"):
+        tool_input = gr.Textbox(label="اكتب طلبك هنا")
+        tool_output = gr.Textbox(label="النتيجة", lines=10)
+        run_btn = gr.Button("تشغيل")
+        
+        def process_tool(input_text):
+            try:
+                tool_call = tool_agent.generate(input_text, TOOLS)
+                if tool_call["tool_name"] == "search_quran":
+                    return quran.search(
+                        tool_call["parameters"]["query"],
+                        tool_call["parameters"]["max_results"]
+                    )
+                elif tool_call["tool_name"] == "generate_text":
+                    return text_gen.generate(
+                        tool_call["parameters"]["prompt"],
+                        tool_call["parameters"]["length"]
+                    )
+                return "لم يتم التعرف على الأداة"
+            except Exception as e:
+                return f"خطأ: {str(e)}"
         
-        tool_btn.click(
-            run_tool_agent,
+        run_btn.click(
+            lambda x: format_output(process_tool(x)),
             inputs=tool_input,
             outputs=tool_output
         )