Update app.py

app.py

@@ -787,11 +787,7 @@ def get_current_time_and_date():
 
 current_time_and_date = get_current_time_and_date()
 
-
-
 def fetch_local_events():
-   
-    
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_events&q=Events+in+Birmingham&hl=en&gl=us&api_key={api_key}'
     response = requests.get(url)
@@ -854,9 +850,6 @@ def fetch_local_events():
     else:
         return "<p>Failed to fetch local events</p>"
 
-
-
-
 def fetch_local_weather():
     try:
         api_key = os.environ['WEATHER_API']
@@ -950,14 +943,6 @@ event type and description.And also add this Birmingham,AL at the end of each ad
 Question: {question}
 Helpful Answer:"""
 
-
-# template2 = """You are an expert concierge who is helpful and a renowned guide for Birmingham,Alabama. Based on today's weather being a sunny bright day and today's date is 1st july 2024, take the location or address but don't show the location or address on the output prompts. Use the following pieces of context, 
-# memory, and message history, along with your knowledge of perennial events in Birmingham,Alabama, to answer the question at the end. If you don't know the answer, just say "Homie, I need to get more data for this," and don't try to make up an answer. 
-# Keep the answer short and sweet and crisp. Always say "It was my pleasure!" at the end of the answer.
-# {context}
-# Question: {question}
-# Helpful Answer:""""
-
 template2 = """You are an expert concierge who is helpful and a renowned guide for Birmingham,Alabama. Based on today's weather being a sunny bright day and today's date is 16th july 2024, take the location or address but don't show the location or address on the output prompts. Use the following pieces of context, 
 memory, and message history, along with your knowledge of perennial events in Birmingham,Alabama, to answer the question at the end. If you don't know the answer, just say "Homie, I need to get more data for this," and don't try to make up an answer. 
 Keep the answer short ,sweet and crisp and in one shot. Always say "It was my pleasure!" at the end of the answer.
@@ -997,7 +982,6 @@ def initialize_agent_with_prompt(prompt_template):
     )
     return agent
 
-
 def generate_answer(message, choice):
     logging.debug(f"generate_answer called with prompt_choice: {choice}")
     
@@ -1013,8 +997,6 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     return response['output'], addresses
 
-
-
 def bot(history, choice, tts_choice, state):
     if not history:
         return history
@@ -1041,7 +1023,11 @@ 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)
 
+def clear_textbox():
+    return ""
 
+def clear_textbox_after_retrieval(chatbot, chat_input, state):
+    return clear_textbox(), clear_textbox()
 
 def print_like_dislike(x: gr.LikeData):
     print(x.index, x.value, x.liked)
@@ -1068,8 +1054,6 @@ def extract_addresses(response):
 
 all_addresses = []
 
-
-
 def generate_map(location_names):
     global all_addresses
     all_addresses.extend(location_names)
@@ -1084,14 +1068,13 @@ def generate_map(location_names):
         if geocode_result:
             location = geocode_result[0]['geometry']['location']
             folium.Marker(
-                [location['lat'], location['lng']],
+                [location['lat'], 'lng'],
                 tooltip=f"{geocode_result[0]['formatted_address']}"
             ).add_to(m)
     
     map_html = m._repr_html_()
     return map_html
 
-
 def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
@@ -1189,7 +1172,7 @@ def transcribe_function(stream, new_chunk):
         stream = y
     result = pipe_asr({"array": stream, "sampling_rate": sr}, return_timestamps=False)
     full_text = result.get("text", "")
-    return stream, full_text  # Return the transcribed text
+    return stream, full_text
 
 def update_map_with_response(history):
     if not history:
@@ -1198,10 +1181,7 @@ def update_map_with_response(history):
     addresses = extract_addresses(response)
     return generate_map(addresses)
 
-def clear_textbox():
-    return "" 
-
-def show_map_if_details(history,choice):
+def show_map_if_details(history, choice):
     if choice in ["Details", "Conversational"]:
         return gr.update(visible=True), update_map_with_response(history)
     else:
@@ -1334,7 +1314,6 @@ def generate_audio_mars5(text):
         cfg = config_class(**{k: kwargs_dict[k] for k in kwargs_dict if k in config_class.__dataclass_fields__})
         ar_codes, wav_out = mars5.tts(chunk, wav, "", cfg=cfg)
         
-        
         temp_audio_path = os.path.join(tempfile.gettempdir(), f"mars5_audio_{len(audio_segments)}.wav")
         torchaudio.save(temp_audio_path, wav_out.unsqueeze(0), mars5.sr)
         audio_segments.append(AudioSegment.from_wav(temp_audio_path))
@@ -1387,6 +1366,7 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             location_output = gr.HTML()
             retriver_button.click(fn=add_message, inputs=[chatbot, chat_input], outputs=[chatbot, chat_input]).then(
                 fn=bot, inputs=[chatbot, choice, tts_choice, state], outputs=[chatbot, gr.Audio(interactive=False, autoplay=True)], api_name="Ask_Retriever").then(
+                fn=clear_textbox_after_retrieval, inputs=[chatbot, chat_input, state], outputs=[chat_input, chat_input]).then(
                 fn=show_map_if_details, inputs=[chatbot, choice], outputs=[location_output, location_output], api_name="map_finder")
             
             bot_msg = chat_msg.then(bot, [chatbot, choice, tts_choice], [chatbot], api_name="generate_voice_response")
@@ -1398,25 +1378,6 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             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="voice_query_to_text")
 
-            # New components for recording and sending chunks
-            audio_recorder = gr.Audio(sources="microphone", type="numpy", label="Record Audio Chunk")
-            send_chunk_button = gr.Button("Send Chunk")
-            
-            def transcribe_chunk(audio):
-                sr, y = audio
-                y = y.astype(np.float32) / np.max(np.abs(y))
-                result = pipe_asr({"array": y, "sampling_rate": sr}, return_timestamps=False)
-                return result["text"]
-            
-            send_chunk_button.click(fn=transcribe_chunk, inputs=audio_recorder, outputs=chat_input).then(
-                fn=add_message, inputs=[chatbot, chat_input], outputs=[chatbot, chat_input])
-
-        with gr.Column():
-             weather_output = gr.HTML(value=fetch_local_weather())
-             news_output = gr.HTML(value=fetch_local_news())
-             events_output = gr.HTML(value=fetch_local_events())
-
-            
         with gr.Column():
             image_output_1 = gr.Image(value=generate_image(hardcoded_prompt_1), width=400, height=400)
             image_output_2 = gr.Image(value=generate_image(hardcoded_prompt_2), width=400, height=400)
@@ -1426,8 +1387,6 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
             location_output = gr.HTML()
             bot_msg.then(show_map_if_details, [chatbot, choice], [location_output, location_output], api_name="map_finder")
-    
-   
 
 demo.queue()
 demo.launch(share=True)
@@ -1435,11 +1394,3 @@ demo.launch(share=True)
 
 
 
-
-
-
-
-
-
-
-