Update app.py

app.py

@@ -1,38 +1,72 @@
 import gradio as gr
 from tools.arabic_generator import ArabicTextGenerator
 from tools.quran_search import QuranSearchEngine
+from tools.tool_agent import ToolCallingAgent  # <-- New import
 
+# Initialize all models
 text_gen = ArabicTextGenerator()
-quran = QuranSearchEngine()
+quran_searcher = QuranSearchEngine()
+tool_agent = ToolCallingAgent()  # <-- New agent
 
-with gr.Blocks(title="الأدوات العربية") as app:
-    # Tab 1: Your Working Arabic Generator (UNTOUCHED)
-    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
+# Define tools for the agent
+TOOLS = [
+    {
+        "name": "search_quran",
+        "description": "Search the Quran for verses",
+        "parameters": {
+            "query": {"type": "string"},
+            "max_results": {"type": "integer"}
+        }
+    },
+    {
+        "name": "generate_text",
+        "description": "Generate Arabic text",
+        "parameters": {
+            "prompt": {"type": "string"},
+            "length": {"type": "integer"}
+        }
+    }
+]
+
+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"
+
+with gr.Blocks(title="Advanced Arabic Tools") as app:
+    # Existing tabs remain unchanged
+    with gr.Tab("🖊️ مولد النصوص"):
+        # ... (keep your existing Arabic generator UI)
 
-    # Tab 2: Quran Surah Viewer (GUARANTEED TO WORK)
     with gr.Tab("📖 القرآن الكريم"):
-        surah_dropdown = gr.Dropdown(
-            label="اختر سورة",
-            choices=quran.get_surahs(),
-            value=1  # Default to Al-Fatiha
-        )
-        show_btn = gr.Button("عرض السورة")
-        quran_output = gr.Textbox(label="النص القرآني", lines=15)
+        # ... (keep your existing Quran search UI)
+
+    # New Tool Calling Agent tab
+    with gr.Tab("🛠️ الوكيل الذكي"):
+        tool_input = gr.Textbox(label="أدخل طلبك")
+        tool_output = gr.JSON(label="استجابة الأداة")
+        tool_btn = gr.Button("تنفيذ")
+        
+        def run_tool_agent(query):
+            tool_call = tool_agent.generate(query, TOOLS)
+            if "error" in tool_call:
+                return tool_call
+            return execute_tool_call(tool_call)
         
-        show_btn.click(
-            quran.get_surah_text,
-            inputs=surah_dropdown,
-            outputs=quran_output
+        tool_btn.click(
+            run_tool_agent,
+            inputs=tool_input,
+            outputs=tool_output
         )
 
-app.launch()
+if __name__ == "__main__":
+    app.launch()