Update app.py

app.py

@@ -3,22 +3,23 @@ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
 model_id = "deepseek-ai/deepseek-coder-1.3b-instruct"
 
+# تحميل الموديل والتوكنيزر مع trust_remote_code=True لأن الموديل مخصص
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-def generate_text(prompt):
-    # تهيئة prompt بشكل واضح (اختياري)
-    formatted_prompt = f"Instruction: {prompt}\nAnswer:"
-    outputs = pipe(
-        formatted_prompt,
-        max_new_tokens=150,
-        temperature=0.7,
-        top_p=0.9,
-        do_sample=True,
-        pad_token_id=tokenizer.eos_token_id,
-    )
-    return outputs[0]['generated_text']
+def chatbot_response(message, history):
+    # استخدام الموديل لتوليد الرد
+    output = pipe(message, max_new_tokens=150, do_sample=True)[0]['generated_text']
+    history = history + [(message, output)]
+    return history, history
 
-gr.Interface(fn=generate_text, inputs="text", outputs="text", title="DeepSeek Coder (1.3B)").launch()
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="اكتب رسالتك هنا...")
+    state = gr.State([])  # حفظ المحادثة
+
+    msg.submit(chatbot_response, inputs=[msg, state], outputs=[chatbot, state])
+
+demo.launch()