Update app.py

app.py

@@ -19,6 +19,10 @@ 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
+if "disease_name" not in st.session_state:
+    st.session_state.disease_name = None
+
 # Function to predict skin disease
 def predict_skin_disease(image):
     image = image.convert("RGB")
@@ -35,7 +39,7 @@ def predict_skin_disease(image):
 
 # Function to get disease details from Groq API
 def get_disease_info(disease_name):
-    prompt = f"Provide a detailed explanation about the skin disease '{disease_name}', including description of disease, causes, precausions, risk and treatment options."
+    prompt = f"Provide a detailed explanation about the skin disease '{disease_name}', including causes, symptoms, and treatment options."
 
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
@@ -46,6 +50,9 @@ def get_disease_info(disease_name):
 
 # Function to handle chatbot queries
 def chatbot_response(disease_name, user_query):
+    if not disease_name:
+        return "Please upload an image and detect the disease first."
+
     prompt = f"The detected skin disease is '{disease_name}'. {user_query}"
     
     chat_completion = client.chat.completions.create(
@@ -65,11 +72,12 @@ 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_column_width=True)
+    st.image(image, caption="Uploaded Image", use_container_width=True)  # Fixed the warning
 
     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}")
@@ -79,13 +87,9 @@ st.subheader("💬 Ask DermaBot")
 user_query = st.text_input("Ask about the detected disease:")
 
 if st.button("Ask"):
-    if uploaded_image:
-        with st.spinner("Thinking..."):
-            response = chatbot_response(disease_name, user_query)
-            st.write(response)
-    else:
-        st.warning("Please upload an image first.")
+    with st.spinner("Thinking..."):
+        response = chatbot_response(st.session_state.disease_name, user_query)
+        st.write(response)
 
 st.markdown("---")
 st.write("🔍 Powered by **AI & Groq API** | © 2025 DermaBot")
-