Update app.py

app.py

@@ -1,9 +1,27 @@
+import os
+from dotenv import load_dotenv
+from huggingface_hub import InferenceClient
 import gradio as gr
+# تحميل المتغيرات من .env
+load_dotenv()
 
-chatbot = gr.load("models/deepseek-ai/DeepSeek-R1", provider="sambanova")
+client = InferenceClient(
+    provider="sambanova",
+    api_key=os.getenv("HF_API_KEY")  # تأكد من إن مفتاح الـ API موجود في بيئة Space بشكل آمن
+)
 
-def predict(input_text):
-    return chatbot(input_text)
+def chat_fn(message, history):
+    completion = client.chat.completions.create(
+        model="deepseek-ai/DeepSeek-R1",
+        messages=[{"role": "user", "content": message}],
+        max_tokens=512,
+    )
+    return completion.choices[0].message.content
 
-iface = gr.Interface(fn=predict, inputs="text", outputs="text")
-iface.launch()
+chat_interface = gr.ChatInterface(
+    fn=chat_fn,
+    title="RAY AI Chat with DeepSeek (Sambanova)",
+    description="نموذج DeepSeek R1 عبر مزود Sambanova - مخصص لتطبيقات الأشعة والذكاء الطبي.",
+)
+
+chat_interface.launch()