Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 from tools.arabic_generator import ArabicTextGenerator
 from tools.quran_search import QuranSearchEngine
-from tools.tool_agent import ToolCallingAgent  # <-- New import
+from tools.tool_agent import ToolCallingAgent
 
 # Initialize all models
 text_gen = ArabicTextGenerator()
 quran_searcher = QuranSearchEngine()
-tool_agent = ToolCallingAgent()  # <-- New agent
+tool_agent = ToolCallingAgent()
 
 # Define tools for the agent
 TOOLS = [
@@ -43,25 +43,38 @@ def execute_tool_call(tool_call):
     return "Tool not found"
 
 with gr.Blocks(title="Advanced Arabic Tools") as app:
-    # Existing tabs remain unchanged
+    # Tab 1: Arabic Text Generator
     with gr.Tab("🖊️ مولد النصوص"):
-        # ... (keep your existing Arabic generator UI)
+        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
+        )
 
+    # Tab 2: Quran Search
     with gr.Tab("📖 القرآن الكريم"):
-        # ... (keep your existing Quran search UI)
+        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
+        )
 
-    # New Tool Calling Agent tab
+    # Tab 3: Tool Calling Agent
     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)
-        
         tool_btn.click(
             run_tool_agent,
             inputs=tool_input,