Deepti-AI commited on
Commit
0b886a5
·
verified ·
1 Parent(s): df685ef

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -222,14 +222,12 @@ async def get_rag_response(user_message: str):
222
  Context : {context}
223
  User Query: {user_message}'''})
224
  # print("chat_messages",chat_messages)
225
- # response = get_response(query, context, chat_messages)
226
- # chat_messages.append({"role": "assistant", "content": response})
227
  return chat_messages
228
 
229
 
230
  # --- GPT + TTS async generator with smaller buffer like second code ---
231
  async def gpt_tts_stream(prompt: str):
232
- print("started gpt_tts_stream",prompt)
233
  global chat_messages
234
  chat_messages = await get_rag_response(prompt)
235
  # print(chat_messages,"chat_messages after getting RAG response")
@@ -254,7 +252,7 @@ async def gpt_tts_stream(prompt: str):
254
  buffer += delta
255
  if len(buffer) >= BUFFER_SIZE and buffer.endswith((".", "!",",", "?", "\n", ";", ":")):
256
  for audio_chunk in tts_chunk_stream(buffer):
257
- print("chunk",buffer)
258
  yield audio_chunk
259
  buffer = ""
260
 
 
222
  Context : {context}
223
  User Query: {user_message}'''})
224
  # print("chat_messages",chat_messages)
 
 
225
  return chat_messages
226
 
227
 
228
  # --- GPT + TTS async generator with smaller buffer like second code ---
229
  async def gpt_tts_stream(prompt: str):
230
+ # print("started gpt_tts_stream",prompt)
231
  global chat_messages
232
  chat_messages = await get_rag_response(prompt)
233
  # print(chat_messages,"chat_messages after getting RAG response")
 
252
  buffer += delta
253
  if len(buffer) >= BUFFER_SIZE and buffer.endswith((".", "!",",", "?", "\n", ";", ":")):
254
  for audio_chunk in tts_chunk_stream(buffer):
255
+ # print("chunk",buffer)
256
  yield audio_chunk
257
  buffer = ""
258