Update app.py

app.py

@@ -126,18 +126,43 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     return response['output'], addresses
 
+# def bot(history, choice):
+#     if not history:
+#         return history
+#     response, addresses = generate_answer(history[-1][0], choice)
+#     history[-1][1] = ""
+#     for character in response:
+#         history[-1][1] += character
+#         time.sleep(0.05)
+#         yield history
+#     if addresses:
+#         return history, addresses
 def bot(history, choice):
     if not history:
         return history
     response, addresses = generate_answer(history[-1][0], choice)
     history[-1][1] = ""
-    for character in response:
-        history[-1][1] += character
-        time.sleep(0.05)
-        yield history
-    if addresses:
-        return history, addresses
+    audio_segments = []
+    tts = gTTS(response, lang='en')
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as f:
+        tts.save(f.name)
+        temp_audio_path = f.name
+
+    audio_chunks = split_audio(temp_audio_path, len(response))
 
+    for idx, character in enumerate(response):
+        history[-1][1] += character
+        if idx < len(audio_chunks):
+            audio_segments.append(audio_chunks[idx])
+        yield history, audio_segments
+
+def split_audio(audio_path, num_chunks):
+    from pydub import AudioSegment
+    audio = AudioSegment.from_file(audio_path)
+    chunk_length = len(audio) / num_chunks
+    chunks = [audio[i * chunk_length:(i + 1) * chunk_length] for i in range(num_chunks)]
+    return chunks
+    
 def add_message(history, message):
     history.append((message, None))
     return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)
@@ -520,6 +545,63 @@ def generate_image(prompt):
 # Hardcoded prompt for image generation
 hardcoded_prompt = "Useing The top events like 'Summer Art Festival' and current time - 4:07 PM ,Date - 06/17/2024 ,Weather-Sunny Bright Day.Create Highly Visually Compelling High Resolution  and High Quality Photographics Advatizement for 'Toyota'"
 
+# with gr.Blocks(theme='rawrsor1/Everforest') as demo:
+#     with gr.Row():
+#         chatbot = gr.Chatbot([], elem_id="chatbot", bubble_full_width=False)
+#         with gr.Column():
+#             weather_output = gr.HTML(value=fetch_local_weather())
+        
+#         with gr.Column():
+#             news_output = gr.HTML(value=fetch_local_news())
+
+#     def setup_ui():
+#         state = gr.State()
+#         with gr.Row():
+#             with gr.Column():
+#               gr.Markdown("<h1>Choose the prompt</h1>", elem_id="prompt-markdown")
+#               choice = gr.Radio(label="Choose a prompt", choices=["Details", "Conversational"], value="Details")
+#             with gr.Column():  # Larger scale for the right column
+#                 gr.Markdown("<h1>Enter the query / Voice Output</h1>", elem_id="query-markdown")
+
+#                 chat_input = gr.Textbox(show_copy_button=True, interactive=True, show_label=False, label="Transcription")
+#                 chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
+#                 bot_msg = chat_msg.then(bot, [chatbot, choice], chatbot, api_name="bot_response")
+#                 bot_msg.then(lambda: gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False), None, [chat_input])
+#                 chatbot.like(print_like_dislike, None, None)
+#                 clear_button = gr.Button("Clear")
+#                 clear_button.click(fn=clear_textbox, inputs=None, outputs=chat_input)
+        
+#             with gr.Column():  # Smaller scale for the left column
+#                 gr.Markdown("<h1>Stream your Voice</h1>", elem_id="voice-markdown")
+
+#                 audio_input = gr.Audio(sources=["microphone"], streaming=True, type='numpy')
+#                 audio_input.stream(transcribe_function, inputs=[state, audio_input], outputs=[state, chat_input], api_name="SAMLOne_real_time")
+
+#         with gr.Row():
+#             with gr.Column():
+#                 gr.Markdown("<h1>Locate the Events</h1>", elem_id="location-markdown")
+#                 location_output = gr.HTML()
+#                 bot_msg.then(show_map_if_details, [chatbot, choice], [location_output, location_output])
+#             with gr.Column():
+#                 gr.Markdown("<h1>Listen the audio</h1>", elem_id="audio-markdown")
+#                 audio_output = gr.Audio()
+#                 #bot_msg.then(generate_audio_elevenlabs, chatbot, audio_output)
+#                 bot_msg_audio = bot_msg.then(lambda history: generate_audio_elevenlabs(history[-1][1]), chatbot, audio_output)
+#         with gr.Row():   
+            
+#             with gr.Column():
+#                 gr.Markdown("<h1>Local Events</h1>", elem_id="events-markdown")
+#                 news_output = gr.HTML(value=fetch_local_events())
+                     
+#             with gr.Column():
+#                 gr.Markdown("<h1>Generated Image</h1>", elem_id="image-markdown")
+#                 image_output = gr.Image(value=generate_image(hardcoded_prompt))
+
+#     setup_ui()    
+
+# demo.queue()
+# demo.launch(share=True)
+
 with gr.Blocks(theme='rawrsor1/Everforest') as demo:
     with gr.Row():
         chatbot = gr.Chatbot([], elem_id="chatbot", bubble_full_width=False)