Update icd9_ui.py

icd9_ui.py

@@ -288,27 +288,27 @@
     # return predictions_with_desc
 
 # Streamlit UI
-st.title("Medical Diagnosis Assistant")
-option = st.selectbox(
-    "Choose Diagnosis Method",
-    ("ICD-9 Code Prediction", "Medical Image Analysis")
-)
-
-# ICD-9 Code Prediction
-if option == "ICD-9 Code Prediction":
-    st.write("### Enter Medical Summary")
-    input_text = st.text_area("Medical Summary", placeholder="Enter clinical notes here...")
-
-    threshold = st.slider("Prediction Threshold", 0.0, 1.0, 0.5, 0.01)
-
-    if st.button("Predict ICD-9 Codes"):
-        if input_text.strip():
-            predictions = predict_icd9([input_text], tokenizer, model, threshold)
-            st.write("### Predicted ICD-9 Codes and Descriptions")
-            for code, description in predictions[0]:
-                st.write(f"- {code}: {description}")
-        else:
-            st.error("Please enter a medical summary.")
+# st.title("Medical Diagnosis Assistant")
+# option = st.selectbox(
+#     "Choose Diagnosis Method",
+#     ("ICD-9 Code Prediction", "Medical Image Analysis")
+# )
+
+# # ICD-9 Code Prediction
+# if option == "ICD-9 Code Prediction":
+#     st.write("### Enter Medical Summary")
+#     input_text = st.text_area("Medical Summary", placeholder="Enter clinical notes here...")
+
+#     threshold = st.slider("Prediction Threshold", 0.0, 1.0, 0.5, 0.01)
+
+#     if st.button("Predict ICD-9 Codes"):
+#         if input_text.strip():
+#             predictions = predict_icd9([input_text], tokenizer, model, threshold)
+#             st.write("### Predicted ICD-9 Codes and Descriptions")
+#             for code, description in predictions[0]:
+#                 st.write(f"- {code}: {description}")
+#         else:
+#             st.error("Please enter a medical summary.")
 
 # Medical Image Analysis
 # elif option == "Medical Image Analysis":
@@ -402,7 +402,7 @@ if option == "ICD-9 Code Prediction":
         # else:
         #     st.info("👆 Please upload a medical image to begin analysis")
 
-        import os
+import os
 import torch
 import pandas as pd
 import streamlit as st