Sksksss commited on
Commit
8e3f795
·
verified ·
1 Parent(s): eb5ebf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -14
app.py CHANGED
@@ -1,22 +1,31 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # نستخدم نموذج جاهز من Hugging Face للإجابة النصية
5
- chatbot = pipeline("text-generation", model="gpt2")
6
 
7
- # دالة المحادثة
8
- def ask_bot(question):
9
- response = chatbot(question, max_length=200, num_return_sequences=1, do_sample=True)
10
- return response[0]["generated_text"]
 
 
 
 
11
 
12
- # واجهة Gradio
13
- demo = gr.Interface(
14
- fn=ask_bot,
15
- inputs=gr.Textbox(label="✍️ اكتب سؤالك هنا"),
16
- outputs=gr.Textbox(label="🤖 جواب البوت"),
17
- title="شات بوت ذكي",
18
- description="تطبيق محادثة بسيط يعمل بالذكاء الاصطناعي للإجابة على أي سؤال.\n\n👨‍💻 المطور: حسين محمد"
19
- )
 
 
 
 
 
20
 
21
  if __name__ == "__main__":
22
  demo.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # نستخدم نموذج محادثة مخصص (يدعم الحوار أفضل من gpt2)
5
+ chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium")
6
 
7
+ # نخزن المحادثة
8
+ def chat(history, user_message):
9
+ from transformers import Conversation
10
+ conv = Conversation(user_message)
11
+ chatbot(conv)
12
+ bot_reply = conv.generated_responses[-1]
13
+ history.append((user_message, bot_reply))
14
+ return history, ""
15
 
16
+ # واجهة Gradio (شات مثل واتساب)
17
+ with gr.Blocks() as demo:
18
+ gr.Markdown("## 🤖 شات بوت عربي تجريبي")
19
+ gr.Markdown("👨‍💻 المطوّر: **حسين محمد**")
20
+
21
+ chatbot_ui = gr.Chatbot()
22
+ msg = gr.Textbox(label="اكتب رسالتك")
23
+ clear = gr.Button("مسح المحادثة")
24
+
25
+ state = gr.State([])
26
+
27
+ msg.submit(chat, [state, msg], [chatbot_ui, msg])
28
+ clear.click(lambda: ([], ""), None, [chatbot_ui, msg])
29
 
30
  if __name__ == "__main__":
31
  demo.launch()