Update app.py

app.py

@@ -1,14 +1,61 @@
+import os
 import gradio as gr
+from openai import OpenAI
 
-def hello(message):
-    return f"Hadithi is working. You said: {message}"
+HF_TOKEN = os.environ.get("HF_TOKEN")
+MODEL_ID = "Qwen/Qwen2.5-7B-Instruct"
 
-demo = gr.Interface(
-    fn=hello,
-    inputs=gr.Textbox(label="Ask"),
-    outputs=gr.Textbox(label="Reply"),
-    title="Hadithi",
-    description="First test for the Hugging Face Space"
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN is missing. Add it in Space Settings -> Secrets.")
+
+client = OpenAI(
+    base_url="https://router.huggingface.co/v1",
+    api_key=HF_TOKEN,
+)
+
+SYSTEM_PROMPT = """
+You are Hadithi AI, a helpful assistant.
+Answer clearly, briefly, and in the same language as the user when possible.
+If the user writes in Arabic, answer in Arabic.
+If the user writes in English, answer in English.
+""".strip()
+
+
+def chat(message, history):
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+
+    messages.append({"role": "user", "content": message})
+
+    try:
+        response = client.chat.completions.create(
+            model=MODEL_ID,
+            messages=messages,
+            temperature=0.3,
+            max_tokens=600,
+        )
+        answer = response.choices[0].message.content.strip()
+    except Exception as e:
+        answer = f"Error: {str(e)}"
+
+    return answer
+
+
+demo = gr.ChatInterface(
+    fn=chat,
+    title="Hadithi AI",
+    description="Chat with Qwen2.5-7B-Instruct",
+    textbox=gr.Textbox(
+        placeholder="Ask anything here...",
+        container=True,
+        scale=7,
+    ),
 )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()