fallahs771 commited on
Commit
5cc362f
·
verified ·
1 Parent(s): ab4f9cd

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +82 -0
app.py ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ import torch
4
+ from TTS.api import TTS
5
+ import os
6
+
7
+ # لود کردن مدل‌ها
8
+ device = "cuda" if torch.cuda.is_available() else "cpu"
9
+
10
+ # مدل تشخیص صدا (اختیاری - می‌تونی از Whisper استفاده کنی)
11
+ # asr = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
12
+
13
+ # مدل هوش مصنوعی فارسی
14
+ chat_model = pipeline(
15
+ "text-generation",
16
+ model="Qwen/Qwen2.5-72B-Instruct",
17
+ device=device,
18
+ max_new_tokens=500
19
+ )
20
+
21
+ # مدل تبدیل متن به صدا با صدای فارسی
22
+ tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
23
+
24
+ def respond(message, history):
25
+ """پاسخ‌دهی هوشمند"""
26
+
27
+ # ساخت پرامپت برای مدل
28
+ prompt = f"""تو یک دستیار هوشمند فارسی‌زبان به نام "دلسا" هستی.
29
+ مهربان، باهوش و کمک‌کننده هستی.
30
+
31
+ کاربر: {message}
32
+ دلسا:"""
33
+
34
+ # دریافت پاسخ از مدل
35
+ response = chat_model(prompt)[0]['generated_text']
36
+
37
+ # استخراج پاسخ دلسا
38
+ ai_response = response.split("دلسا:")[-1].strip()
39
+
40
+ # تبدیل متن به صدا
41
+ audio_file = tts.tts(
42
+ text=ai_response,
43
+ speaker_wav="persian_female_voice.wav", # نمونه صدای فارسی زنانه
44
+ language="fa"
45
+ )
46
+
47
+ return ai_response, audio_file
48
+
49
+ # ساخت رابط گرافیکی
50
+ with gr.Blocks(
51
+ theme=gr.themes.Soft(primary_hue="cyan"),
52
+ css="""
53
+ .container {
54
+ background: linear-gradient(135deg, #0f0c29, #302b63, #24243e);
55
+ min-height: 100vh;
56
+ }
57
+ .chatbot {
58
+ border-radius: 20px;
59
+ box-shadow: 0 0 30px rgba(0, 255, 255, 0.3);
60
+ }
61
+ """
62
+ ) as demo:
63
+
64
+ gr.Markdown(
65
+ """
66
+ # 🌟 دلسا - دستیار هوشمند کهکشانی
67
+ ### با من صحبت کن، من به فارسی جواب میدم! ✨
68
+ """
69
+ )
70
+
71
+ chat_interface = gr.ChatInterface(
72
+ fn=respond,
73
+ title="دلسا",
74
+ description="هوش مصنوعی فارسی‌زبان با صدای طبیعی",
75
+ examples=[
76
+ "سلام دلسا، حالت چطوره؟",
77
+ "یه شعر زیبا برام بگو",
78
+ "در مورد هوش مصنوعی توضیح بده"
79
+ ]
80
+ )
81
+
82
+ demo.launch()