Shroukkkk commited on
Commit
55c5783
·
verified ·
1 Parent(s): a88e931

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -153,8 +153,9 @@ QWEN_MODEL_ID = os.getenv("QWEN_MODEL_ID", "Qwen/Qwen2.5-3B-Instruct")
153
  WHISPER_SIZE = os.getenv("WHISPER_SIZE", "small")
154
  XTTS_MODEL_ID = os.getenv("XTTS_MODEL_ID", "tts_models/multilingual/multi-dataset/xtts_v2")
155
 
 
156
  SYSTEM_PROMPT = os.getenv(
157
- "SYSTEM_PROMPT"
158
  "انت مساعد مصري. رد باللهجة المصرية فقط وبالعربي فقط. ممنوع تستخدم اي كلام انجليزي. ردودك قصيرة وواضحة."
159
  )
160
 
@@ -205,11 +206,19 @@ def transcribe_file(path: str) -> str:
205
  # history format: list of [user, assistant]
206
  # ----------------------------
207
  def qwen_reply(history, user_text: str) -> str:
208
- messages = [{"role": "system", "content": SYSTEM_PROMPT}]
209
 
210
  for u, a in (history or [])[-MAX_TURNS:]:
211
- messages.append({"role": "user", "content": u})
212
- messages.append({"role": "assistant", "content": a})
 
 
 
 
 
 
 
 
213
 
214
  messages.append({"role": "user", "content": user_text})
215
 
 
153
  WHISPER_SIZE = os.getenv("WHISPER_SIZE", "small")
154
  XTTS_MODEL_ID = os.getenv("XTTS_MODEL_ID", "tts_models/multilingual/multi-dataset/xtts_v2")
155
 
156
+
157
  SYSTEM_PROMPT = os.getenv(
158
+ "SYSTEM_PROMPT",
159
  "انت مساعد مصري. رد باللهجة المصرية فقط وبالعربي فقط. ممنوع تستخدم اي كلام انجليزي. ردودك قصيرة وواضحة."
160
  )
161
 
 
206
  # history format: list of [user, assistant]
207
  # ----------------------------
208
  def qwen_reply(history, user_text: str) -> str:
209
+ messages = [{"role": "system", "content": SYSTEM_PROMPT or ""}]
210
 
211
  for u, a in (history or [])[-MAX_TURNS:]:
212
+ u = "" if u is None else str(u)
213
+ a = "" if a is None else str(a)
214
+ if u.strip():
215
+ messages.append({"role": "user", "content": u})
216
+ if a.strip():
217
+ messages.append({"role": "assistant", "content": a})
218
+
219
+ user_text = "" if user_text is None else str(user_text)
220
+ if not user_text.strip():
221
+ return "ممكن تكتب سؤالك تاني؟"
222
 
223
  messages.append({"role": "user", "content": user_text})
224