Update app.py

app.py

@@ -28,10 +28,16 @@ def process_audio(audio_bytes):
     return {"raw": waveform.numpy().squeeze(), "sampling_rate": 16000}
 
 # Voice input component
+# Updated voice_input function
 def voice_input(key, prompt_text, default_text=""):
+    # Initialize session state keys if they don't exist
+    if f"text_{key}" not in st.session_state:
+        st.session_state[f"text_{key}"] = default_text
+    
     col1, col2 = st.columns([4, 1])
     with col1:
-        text_input = st.text_input(prompt_text, value=default_text, key=f"text_{key}")
+        # Create the text input with the current session state value
+        text_value = st.text_input(prompt_text, value=st.session_state[f"text_{key}"], key=f"text_input_{key}")
     with col2:
         audio_bytes = audio_recorder(
             pause_threshold=0.8,
@@ -51,15 +57,15 @@ def voice_input(key, prompt_text, default_text=""):
                 whisper = load_model()
                 transcribed_text = whisper(audio_input)["text"]
                 
-                # Update the corresponding text input
+                # Update the session state value (this happens before widget creation)
                 st.session_state[f"text_{key}"] = transcribed_text
                 st.rerun()
                 
             except Exception as e:
                 st.error(f"Error in voice input: {str(e)}")
     
-    return text_input
-
+    # Return the current text value (from either manual input or voice)
+    return st.session_state[f"text_{key}"]
 # Enhanced Custom CSS with modern design
 def inject_custom_css():
     st.markdown("""