Update main.py
Browse files
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 |
|