Update app.py

app.py

@@ -19,9 +19,11 @@ processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
 GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")  
 client = Groq(api_key=GROQ_API_KEY)
 
-# Initialize session state for disease_name
+# Initialize session state for disease details
 if "disease_name" not in st.session_state:
     st.session_state.disease_name = None
+if "disease_info" not in st.session_state:
+    st.session_state.disease_info = None
 
 # Function to predict skin disease
 def predict_skin_disease(image):
@@ -67,23 +69,31 @@ st.image("https://huggingface.co/spaces/your-huggingface-space/logo.png", width=
 st.title("🩺 DermaBot - AI Skin Disease Detector")
 st.write("Upload an image of a skin condition to get a diagnosis and ask questions about it.")
 
-# Upload image section
+# Step 1: Upload image
 uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
 if uploaded_image:
     image = Image.open(uploaded_image)
-    st.image(image, caption="Uploaded Image", use_container_width=True)  # Fixed the warning
+    st.image(image, caption="Uploaded Image", use_container_width=True)
 
+    # Step 2: Detect disease
     if st.button("Detect Disease"):
         with st.spinner("Analyzing..."):
             disease_name = predict_skin_disease(image)
-            st.session_state.disease_name = disease_name  # Store in session state
             disease_info = get_disease_info(disease_name)
-            st.success(f"**Detected Disease:** {disease_name}")
-            st.write(f"**Details:** {disease_info}")
 
-# Chatbot section
-st.subheader("💬 Ask DermaBot")
+            # Store results in session state
+            st.session_state.disease_name = disease_name
+            st.session_state.disease_info = disease_info
+
+# Display detected disease information if available
+if st.session_state.disease_name:
+    st.success(f"**Detected Disease:** {st.session_state.disease_name}")
+    st.write(f"**Details:** {st.session_state.disease_info}")
+
+# Step 3: Chatbot
+st.subheader("💬 Ask DermaBot about this disease")
+
 user_query = st.text_input("Ask about the detected disease:")
 
 if st.button("Ask"):