Update app.py

app.py

@@ -1,68 +1,48 @@
 import os
 import gradio as gr
-import requests
-import json
+from deepsea import DeepSeaClient  
 
-API_KEY = os.getenv("HF_API_KEY")
-MODEL_NAME = "mistralai/Mixtral-8x7B-Instruct-v0.1"  
+API_KEY = os.environ.get("DEEPSEA_API_KEY")
+if not API_KEY:
+    raise ValueError("الرجاء وضع DEEPSEA_API_KEY في البيئة (Secrets على Hugging Face)")
 
-def generate_main_question_mistral(paragraph: str):
-    if not paragraph or paragraph.strip() == "":
+client = DeepSeaClient(api_key=API_KEY)
+MODEL_NAME = "text-generator-arabic"  
+
+def generate_main_question_deepsea(paragraph: str):
+    if not paragraph.strip():
         return "الرجاء إدخال فقرة أولاً."
 
     prompt = f"""
-    الفقرة التالية:
-    {paragraph}
-
-    المطلوب:
-    اقرأ الفقرة السابقة،
-    اعتمادًا عليها قم بتوليد سؤال أساسي للفقرة ليكون مفتاحًا لعملية تسميع الطالب للفقرة.
-    اكتب السؤال المولد فقط بدون شرح ولا أي تفسير.
-    """
-
-    payload = {
-        "model": MODEL_NAME,
-        "prompt": prompt,
-        "max_tokens": 150,
-        "temperature": 0.3
-    }
+الفقرة التالية:
+{paragraph}
 
-    headers = {
-        "Authorization": f"Bearer {API_KEY}",
-        "Content-Type": "application/json"
-    }
+المطلوب:
+اعتمادًا على الفقرة السابقة، قم بتوليد سؤال أساسي للفقرة ليكون مفتاحًا لعملية تسميع الطالب.
+اكتب السؤال فقط بدون أي شرح أو تفسير.
+"""
 
     try:
-        response = requests.post(
-            "https://router.huggingface.co/v1/completions",
-            headers=headers,
-            data=json.dumps(payload)
-        )
-
-        if response.status_code != 200:
-            return f"Error while connecting to API: {response.text}"
-
-        result = response.json()
-        return result["choices"][0]["text"].strip()
-
+        response = client.generate_text(model=MODEL_NAME, prompt=prompt, max_tokens=150)
+        return response.text.strip()
     except Exception as e:
-        return f"Error while connecting to API: {e}"
-
+        return f"Error while connecting to DeepSea API: {e}"
 
+# واجهة Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## MainQuestion — Basic Question Generator (Arabic Output)")
-
+    
     with gr.Row():
         paragraph = gr.Textbox(
-            label="Paragraph (Input text)",
-            lines=8,
+            label="Paragraph (Input text)", 
+            lines=8, 
             placeholder="Paste the paragraph here..."
         )
-
+    
     output = gr.Textbox(label="Generated Question (Arabic)", lines=3)
 
     submit_btn = gr.Button("Submit")
-    submit_btn.click(fn=generate_main_question_mistral, inputs=paragraph, outputs=output)
+    submit_btn.click(fn=generate_main_question_deepsea, inputs=paragraph, outputs=output)
 
 if __name__ == "__main__":
     demo.launch(share=True, show_error=True)