Update app.py

app.py

@@ -45,6 +45,12 @@ if 'chat_history' not in st.session_state:
     st.session_state['chat_history'] = []
 if 'file_uploader_key' not in st.session_state:
     st.session_state['file_uploader_key'] = str(uuid.uuid4())
+if 'uploaded_files' not in st.session_state:
+    st.session_state['uploaded_files'] = []
+if 'user_input' not in st.session_state:
+    st.session_state['user_input'] = ''
+if 'audio_data' not in st.session_state:
+    st.session_state['audio_data'] = None
 
 # --- Streamlit UI ---
 st.title("Gemini Chatbot")
@@ -64,6 +70,9 @@ def get_file_base64(file_content, mime_type):
 def clear_conversation():
     st.session_state['chat_history'] = []
     st.session_state['file_uploader_key'] = str(uuid.uuid4())
+    st.session_state['user_input'] = ''
+    st.session_state['uploaded_files'] = []
+    st.session_state['audio_data'] = None
 
 def display_chat_history():
     chat_container = st.empty()
@@ -172,7 +181,6 @@ uploaded_files = st.file_uploader(
 # --- WebRTC Audio Recording ---
 RTC_CONFIGURATION = RTCConfiguration({"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]})
 
-# Run webrtc_streamer in a separate async function to avoid conflicts
 async def run_webrtc():
     webrtc_ctx = webrtc_streamer(
         key="audio-recorder",
@@ -185,10 +193,10 @@ async def run_webrtc():
     if webrtc_ctx.audio_receiver:
         st.write("Recording audio...")
         audio_frames = webrtc_ctx.audio_receiver.get_frames(timeout=None)
-        audio_data = b"".join([frame for frame in audio_frames])
+        st.session_state.audio_data = b"".join([frame.to_ndarray() for frame in audio_frames])
 
         if st.button("Send Recording"):
-            send_message(audio_data=audio_data)
+            send_message(audio_data=st.session_state.audio_data)
 
 # --- Other Buttons ---
 st.button("Clear Conversation", on_click=clear_conversation)