Update app.py

app.py

@@ -2,16 +2,27 @@ import streamlit as st
 import os
 import numpy as np
 import matplotlib.pyplot as plt
-import torch
-from datetime import datetime, timedelta
 from transformers import pipeline
+from datetime import datetime, timedelta
+
+# Load API Key securely
+GEMINI_API_KEY = st.secrets["Gemini_API_Key"] if "Gemini_API_Key" in st.secrets else os.getenv("Gemini_API_Key")
 
 # Set Up Multi-Page Navigation
 st.set_page_config(page_title="OncoPlan", page_icon="⚕️", layout="wide")
 
+# Aesthetic Title Section
+st.markdown(
+    """
+    # ⚕️ OncoPlan
+    ### "Where Science Meets Compassion – AI-Powered Personalized Radiotherapy Planning."
+    #### Beyond Technology, Towards a Healthier Tomorrow.
+    """, unsafe_allow_html=True
+)
+
 # Sidebar Navigation
 st.sidebar.header("🔍 Navigation")
-page_options = ["🏠 Home", "📋 Patient Info & Tumor Details", "📈 Progress Charts", "🤖 AI Chatbot"]
+page_options = ["🏠 Home", "📋 Patient Info & Tumor Details", "💊 Treatment Plan", "📈 Progress Charts", "🤖 AI Chatbot"]
 
 if "current_page_index" not in st.session_state:
     st.session_state.current_page_index = 0  # Track current page index
@@ -22,9 +33,10 @@ def set_page(index):
 
 # Cancer Types & Regions
 cancer_types = {
-    "Brain Tumors": ["Frontal Lobe", "Parietal Lobe", "Temporal Lobe", "Occipital Lobe", "Cerebellum"],
-    "Breast Cancer": ["Left Breast", "Right Breast", "Ductal Region", "Axillary Lymph Nodes"],
-    "Lung Cancer": ["Left Lung", "Right Lung", "Mediastinum", "Pleura"],
+    "Brain Tumors": ["Left Cerebral Hemisphere", "Right Cerebral Hemisphere", "Frontal Lobe", "Parietal Lobe", "Temporal Lobe", "Occipital Lobe", "Brainstem", "Cerebellum"],
+    "Breast Cancer": ["Left Breast", "Right Breast", "Ductal Region", "Lobular Region", "Axillary Lymph Nodes"],
+    "Lung Cancer": ["Left Lung - Upper Lobe", "Left Lung - Lower Lobe", "Right Lung - Upper Lobe", "Right Lung - Middle Lobe", "Right Lung - Lower Lobe", "Mediastinum", "Pleura"],
+    "Prostate Cancer": ["Peripheral Zone", "Transitional Zone", "Central Zone", "Seminal Vesicles", "Urethral Region"],
     "Other": ["Custom"]
 }
 
@@ -53,6 +65,7 @@ if page == "🏠 Home":
         
         ### 🔹 Features:
         - 📋 **Patient & Tumor Data Entry**
+        - 💊 **AI-Generated Treatment Plans**
         - 📈 **Progress Visualization**
         - 🤖 **AI Chatbot for Medical Guidance**
     """)
@@ -69,14 +82,22 @@ elif page == "📋 Patient Info & Tumor Details":
     st.write(f"### Patient: {st.session_state.get('name', 'Unknown')} ({st.session_state.get('age', 'N/A')} years old)")
 
     # Cancer Type & Region
-    selected_cancer_type = st.selectbox("Cancer Type", list(cancer_types.keys()), index=0)
-    selected_cancer_region = st.selectbox("Cancer Region", cancer_types[selected_cancer_type])
+    cancer_type = st.session_state.get("cancer_type", "Brain Tumors")
+    cancer_region_options = cancer_types[cancer_type]
+    default_region = st.session_state.get("cancer_region", cancer_region_options[0])
+
+    if default_region not in cancer_region_options:
+        default_region = cancer_region_options[0]
+
+    st.session_state["cancer_type"] = st.selectbox("Cancer Type", list(cancer_types.keys()), index=list(cancer_types.keys()).index(cancer_type))
+    st.session_state["cancer_region"] = st.selectbox("Cancer Region", cancer_region_options, index=cancer_region_options.index(default_region))
 
     # Tumor Size
-    tumor_size = (
-        st.number_input("Tumor Length (mm)", min_value=1, step=1, value=10),
-        st.number_input("Tumor Width (mm)", min_value=1, step=1, value=10),
-        st.number_input("Tumor Height (mm)", min_value=1, step=1, value=10)
+    tumor_size = st.session_state.get("tumor_size", (10, 10, 10))
+    st.session_state["tumor_size"] = (
+        st.number_input("Tumor Length (mm)", min_value=1, step=1, value=tumor_size[0]),
+        st.number_input("Tumor Width (mm)", min_value=1, step=1, value=tumor_size[1]),
+        st.number_input("Tumor Height (mm)", min_value=1, step=1, value=tumor_size[2])
     )
     st.success("Tumor details saved!")
 
@@ -87,9 +108,11 @@ elif page == "📈 Progress Charts":
     st.subheader("📊 Treatment Progress Chart")
     treatment_sessions = list(range(1, 31))
     total_dose = [i * 2 for i in treatment_sessions]
-    
+    skin_dose = [i * 1.5 for i in treatment_sessions]
+
     fig, ax = plt.subplots()
     ax.plot(treatment_sessions, total_dose, label="Total Target Dose (Gy)", marker="o")
+    ax.plot(treatment_sessions, skin_dose, label="Skin Dose (Gy)", marker="s", linestyle="--")
     ax.set_xlabel("Session #")
     ax.set_ylabel("Dose (Gy)")
     ax.legend()
@@ -101,15 +124,10 @@ elif page == "📈 Progress Charts":
 # AI Chatbot
 elif page == "🤖 AI Chatbot":
     st.subheader("💬 AI Chatbot")
-    
-    # Use Hugging Face's OpenAI-compatible API for text generation
+    chatbot = pipeline("text-generation", model="gpt2")
     user_input = st.text_input("Ask your question:")
-    
     if user_input:
-        chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", device="cpu")
         response = chatbot(user_input, max_length=100)
         st.write("🤖 AI Response:", response[0]['generated_text'])
 
     navigation_buttons()
-
-