Update src/streamlit_app.py

src/streamlit_app.py

@@ -93,19 +93,24 @@ for tab, sig in zip(tabs, ["ECG", "EMG", "VAG", "PCG"]):
                     "_Note: Do not upload `.mp3`, `.flac`, or stereo files—they may fail to process properly._"
                 )
 
+# Replace your file uploader section with this corrected version:
+
+        # Initialize session state
         if f"uploaded_file_{sig}" not in st.session_state:
             st.session_state[f"uploaded_file_{sig}"] = None
-        if f"file_processed_{sig}" not in st.session_state:
-            st.session_state[f"file_processed_{sig}"] = False
-      
+        
+        # File uploader with proper callback
+        def handle_file_upload():
+            st.session_state[f"uploaded_file_{sig}"] = st.session_state[f"upload_{sig}"]
+        
         uploaded = st.file_uploader(
             f"Upload {sig} file",
             type=FILE_TYPES[sig],
-            key=f"upload_{sig}"
-            on_change=lambda: st.session_state.update({
-            f"uploaded_file_{sig}": st.session_state[f"upload_{sig}"],
-            f"file_processed_{sig}": False})
+            key=f"upload_{sig}",
+            on_change=handle_file_upload
         )
+        
+        # Check if file was uploaded
         if uploaded is not None:
             st.session_state[f"uploaded_file_{sig}"] = uploaded
             st.success(f"✅ File uploaded: {uploaded.name} ({uploaded.size} bytes)")
@@ -118,17 +123,15 @@ for tab, sig in zip(tabs, ["ECG", "EMG", "VAG", "PCG"]):
             with col2:
                 st.info(f"📏 **Size:** {uploaded.size} bytes")
                 st.info(f"✅ **Status:** Ready for analysis")
-            
-            # Auto-enable the run button
-            st.session_state[f"file_processed_{sig}"] = True
-            
+        
         elif st.session_state[f"uploaded_file_{sig}"] is not None:
-            # Use previously uploaded file
             uploaded = st.session_state[f"uploaded_file_{sig}"]
-            st.info(f"📁 Using file: {uploaded.name}")
+            st.info(f"📁 Using previously uploaded file: {uploaded.name}")
         else:
             st.warning("⚠️ No file uploaded yet")
             st.info(f"💡 Supported formats: {', '.join(FILE_TYPES[sig])}")
+        
+        # Add control buttons
         col1, col2 = st.columns([1, 1])
         with col1:
             if st.button(f"🔄 Refresh", key=f"refresh_{sig}"):
@@ -138,63 +141,57 @@ for tab, sig in zip(tabs, ["ECG", "EMG", "VAG", "PCG"]):
             if st.session_state[f"uploaded_file_{sig}"] is not None:
                 if st.button(f"🗑️ Clear File", key=f"clear_{sig}"):
                     st.session_state[f"uploaded_file_{sig}"] = None
-                    st.session_state[f"file_processed_{sig}"] = False
                     st.rerun()
         
-        # Update your analysis buttons to use session state
+        # Your existing analysis buttons (keep them the same, but check session state)
         if st.session_state[f"uploaded_file_{sig}"] is not None:
             uploaded = st.session_state[f"uploaded_file_{sig}"]
-               
-                if sig == "ECG" and uploaded and st.button("Run Diagnostic", key=f"run_{sig}", type="primary" ):
-                    label, human, conf, gnote = analyze_signal(
-                        uploaded, MODELS["ECG"], GEMINI_API_KEY, signal_type="ECG"
-                    )
-                    st.success(f"**{label} – {human}**\n\nConfidence: {conf:.2%}")
-                    if gnote:
-                        st.markdown("### 🧠 Gemini Insight")
-                        st.write(gnote)
-                    elif not GEMINI_API_KEY:
-                        st.info("Gemini key missing – no explanation.")
-
-    
-                elif sig == "PCG" and uploaded and st.button("Run Diagnostic", key=f"run_{sig}", type="primary"):
-                    label, human, conf, gnote = analyze_pcg_signal(
-                        uploaded, MODELS["PCG"], GEMINI_API_KEY
-                    )
-                    st.success(f"**{label}**\n\nConfidence: {conf:.2%}")
-                    if gnote:
-                        st.markdown("### 🧠 Gemini Insight")
-                        st.write(gnote)
-                    elif not GEMINI_API_KEY:
-                        st.info("Gemini key missing – no explanation.")
+            
+            if sig == "ECG" and st.button("Run Diagnostic", key=f"run_{sig}"):
+                label, human, conf, gnote = analyze_signal(
+                    uploaded, MODELS["ECG"], GEMINI_API_KEY, signal_type="ECG"
+                )
+                st.success(f"**{label} – {human}**\n\nConfidence: {conf:.2%}")
+                if gnote:
+                    st.markdown("### 🧠 Gemini Insight")
+                    st.write(gnote)
+                elif not GEMINI_API_KEY:
+                    st.info("Gemini key missing – no explanation.")
+        
+            elif sig == "PCG" and st.button("Run Diagnostic", key=f"run_{sig}"):
+                label, human, conf, gnote = analyze_pcg_signal(
+                    uploaded, MODELS["PCG"], GEMINI_API_KEY
+                )
+                st.success(f"**{label}**\n\nConfidence: {conf:.2%}")
+                if gnote:
+                    st.markdown("### 🧠 Gemini Insight")
+                    st.write(gnote)
+                elif not GEMINI_API_KEY:
+                    st.info("Gemini key missing – no explanation.")
         
-                elif sig == "EMG" and uploaded and st.button("Run Diagnostic", key=f"run_{sig}", type="primary"):
-                    human, conf, gnote = analyze_emg_signal(
+            elif sig == "EMG" and st.button("Run Diagnostic", key=f"run_{sig}"):
+                human, conf, gnote = analyze_emg_signal(
                     uploaded, MODELS["EMG"], GEMINI_API_KEY
-                    )
-                    st.success(f"**{human.upper()}**\n\nConfidence: {conf:.2%}")
-                    if gnote:
-                        st.markdown("### 🧠 Gemini Insight")
-                        st.write(gnote)
-                    elif not GEMINI_API_KEY:
-                        st.info("Gemini key missing – no explanation.")
-                        
-                elif sig == "VAG" and uploaded and st.button("Run Diagnostic", key=f"run_{sig}", type="primary"):
+                )
+                st.success(f"**{human.upper()}**\n\nConfidence: {conf:.2%}")
+                if gnote:
+                    st.markdown("### 🧠 Gemini Insight")
+                    st.write(gnote)
+                elif not GEMINI_API_KEY:
+                    st.info("Gemini key missing – no explanation.")
                     
-                    label, human, conf, gnote = predict_vag_from_features(
-                        uploaded, MODELS["VAG"], GEMINI_API_KEY
-                    )
-                    st.success(f"**{label}**\n\nConfidence: {conf:.2%}")
-                    if gnote:
-                        st.markdown("### 🧠 Gemini Insight")
-                        st.write(gnote)
-                    elif not GEMINI_API_KEY:
-                        st.info("Gemini key missing – no explanation.")
-
-
+            elif sig == "VAG" and st.button("Run Diagnostic", key=f"run_{sig}"):
+                label, human, conf, gnote = predict_vag_from_features(
+                    uploaded, MODELS["VAG"], GEMINI_API_KEY
+                )
+                st.success(f"**{label}**\n\nConfidence: {conf:.2%}")
+                if gnote:
+                    st.markdown("### 🧠 Gemini Insight")
+                    st.write(gnote)
+                elif not GEMINI_API_KEY:
+                    st.info("Gemini key missing – no explanation.")
         else:
-            if not uploaded:
-                st.info("Upload a file to begin.")
+            st.info("📤 Upload a file to begin analysis")
 
 
 st.caption("© 2025 Biosignal Chatbot")