Update src/streamlit_app.py

src/streamlit_app.py

@@ -93,103 +93,58 @@ 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
-        
-        # 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=handle_file_upload
+            key=f"upload_{sig}"
         )
-        
-        # 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)")
-            
-            # Show file details
-            col1, col2 = st.columns(2)
-            with col1:
-                st.info(f"📁 **Filename:** {uploaded.name}")
-                st.info(f"🏷️ **Type:** {uploaded.type}")
-            with col2:
-                st.info(f"📏 **Size:** {uploaded.size} bytes")
-                st.info(f"✅ **Status:** Ready for analysis")
-        
-        elif st.session_state[f"uploaded_file_{sig}"] is not None:
-            uploaded = st.session_state[f"uploaded_file_{sig}"]
-            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}"):
-                st.rerun()
-        
-        with col2:
-            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.rerun()
-        
-        # 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}"):
+            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}"):
+            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}"):
+            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}"):
             
-            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 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 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.")
+            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:
             st.info("📤 Upload a file to begin analysis")