Update app.py

app.py

@@ -1,3 +1,5 @@
+#file_path = "cleaned_bmd_medication_data.xlsx"
+
 import streamlit as st
 import pandas as pd
 import plotly.graph_objs as go
@@ -12,8 +14,7 @@ REGRESSION_CONSTANTS = {
 # Load medication data
 @st.cache_data
 def load_medication_data():
-    # Use the combined medication data we cleaned earlier
-    file_path = "cleaned_bmd_medication_data.xlsx"
+    file_path = ""cleaned_bmd_medication_data.xlsx""
     return pd.read_excel(file_path)
 
 # Calculate predicted BMD after medication
@@ -59,6 +60,10 @@ def display_results(predictions, site):
         tscore_predictions = result['T-score Predictions']
         years = ['0', '1st', '2nd', '3rd', '4th', '5th', '6th', '8th', '10th']
         
+        # Format values
+        bmd_predictions = [f"{bmd:.3f}" for bmd in bmd_predictions]
+        tscore_predictions = [f"{tscore:.1f}" for tscore in tscore_predictions]
+        
         # Create table
         data = {
             'Year': years[:len(bmd_predictions)],
@@ -69,13 +74,15 @@ def display_results(predictions, site):
         st.dataframe(pd.DataFrame(data))
         
         # Create plots
-        fig_bmd = go.Figure(data=[go.Scatter(x=data['Year'], y=data['Predicted BMD'], mode='lines+markers', name='BMD')])
-        fig_bmd.update_layout(title=f"{drug} - Predicted BMD over Time", xaxis_title="Year", yaxis_title="BMD (g/cm²)")
-        st.plotly_chart(fig_bmd)
-        
-        fig_tscore = go.Figure(data=[go.Scatter(x=data['Year'], y=data['Predicted T-score'], mode='lines+markers', name='T-score')])
-        fig_tscore.update_layout(title=f"{drug} - Predicted T-score over Time", xaxis_title="Year", yaxis_title="T-score")
-        st.plotly_chart(fig_tscore)
+        fig_bmd = go.Scatter(x=data['Year'], y=[float(bmd) for bmd in bmd_predictions], mode='lines+markers', name='BMD')
+        fig_tscore = go.Scatter(x=data['Year'], y=[float(tscore) for tscore in tscore_predictions], mode='lines+markers', name='T-score')
+
+        # Combine both graphs in a single row
+        col1, col2 = st.columns(2)
+        with col1:
+            st.plotly_chart(go.Figure(data=[fig_bmd], layout=go.Layout(title=f"{drug} - Predicted BMD", xaxis_title="Year", yaxis_title="BMD (g/cm²)")))
+        with col2:
+            st.plotly_chart(go.Figure(data=[fig_tscore], layout=go.Layout(title=f"{drug} - Predicted T-score", xaxis_title="Year", yaxis_title="T-score")))
 
 # Streamlit UI
 def main():