update

app.py

@@ -327,93 +327,94 @@ col1, col2 = st.columns([1, 1])
 with col1:
     st.header(" Upload & Configure")
     
-    with st.form("diagnosis_form"):
-        uploaded_file = st.file_uploader(
-            "Choose a device image",
-            type=['jpg', 'jpeg', 'png'],
-            help="Upload a clear image of your device",
-            key="uploader"
-        )
+    #
+    uploaded_file = st.file_uploader(
+        "Choose a device image",
+        type=['jpg', 'jpeg', 'png'],
+        help="Upload a clear image of your device",
+        key="uploader"
+    )
         
-        if uploaded_file:
-            st.session_state.image_bytes = uploaded_file.getvalue()
-            st.session_state.image_name = uploaded_file.name
+    if uploaded_file:
+        st.session_state.image_bytes = uploaded_file.getvalue()
+        st.session_state.image_name = uploaded_file.name
 
         
-        # Load image from session state (persists across reruns)
-        image = None
-        if "image_bytes" in st.session_state:
-            image = Image.open(io.BytesIO(st.session_state.image_bytes))
-            st.image(image, caption="Uploaded Image", use_column_width=False)
+    # Load image from session state (persists across reruns)
+    image = None
+    if "image_bytes" in st.session_state:
+        image = Image.open(io.BytesIO(st.session_state.image_bytes))
+        st.image(image, caption="Uploaded Image", use_column_width=False)
         
-        description = st.text_area(
-            "Describe the issue (optional)",
-            placeholder="e.g., My phone screen cracked after I dropped it. Touch is not working properly.",
-            height=100,
-            help="Providing a description improves accuracy"
-        )
+    description = st.text_area(
+        "Describe the issue (optional)",
+        placeholder="e.g., My phone screen cracked after I dropped it. Touch is not working properly.",
+        height=100,
+        help="Providing a description improves accuracy"
+    )
         
-        st.subheader(" Device Information (for price prediction)")
+    st.subheader(" Device Information (for price prediction)")
         
-        col_a, col_b = st.columns(2)
+    col_a, col_b = st.columns(2)
         
-        with col_a:
-            brand = st.selectbox(
-                "Brand",
-                ["Apple", "Samsung", "OnePlus", "Dell", "HP", "Lenovo", "Other"]
-            )
+    with col_a:
+        brand = st.selectbox(
+            "Brand",
+            ["Apple", "Samsung", "OnePlus", "Dell", "HP", "Lenovo", "Other"]
+        )
             
-            device_type = st.selectbox(
-                "Device Type",
-                ["Phone", "Laptop"]
-            )
+        device_type = st.selectbox(
+            "Device Type",
+            ["Phone", "Laptop"]
+        )
         
-        model_options = {
-            "Apple": {
-                "Phone": ["iPhone 15 Pro Max", "iPhone 15 Pro", "iPhone 15", "iPhone 14 Pro Max", "iPhone 14 Pro", "iPhone 14", "iPhone 13 Pro Max", "iPhone 13 Pro", "iPhone 13", "iPhone 12 Pro Max", "iPhone 12 Pro", "iPhone 12", "iPhone 11 Pro Max", "iPhone 11 Pro", "iPhone 11", "iPhone XS Max", "iPhone XS", "iPhone XR", "iPhone X", "iPhone 8 Plus", "iPhone 8"],
-                "Laptop": ["MacBook Pro 16\" M3 Max", "MacBook Pro 14\" M3 Pro", "MacBook Pro 16\" M2 Max", "MacBook Pro 14\" M2 Pro", "MacBook Air M3", "MacBook Air M2", "MacBook Air M1", "MacBook Pro 13\" M1"]
-            },
-            "Samsung": {
-                "Phone": ["Galaxy S24 Ultra", "Galaxy S24+", "Galaxy S24", "Galaxy S23 Ultra", "Galaxy S23+", "Galaxy S23", "Galaxy S22 Ultra", "Galaxy S22+", "Galaxy S22", "Galaxy Z Fold 5", "Galaxy Z Flip 5", "Galaxy Z Fold 4", "Galaxy Z Flip 4", "Galaxy A54", "Galaxy A34"],
-                "Laptop": ["Galaxy Book4 Pro", "Galaxy Book3 Ultra", "Galaxy Book3 Pro 360", "Galaxy Book2 Pro", "Galaxy Book2"]
+    model_options = {
+        "Apple": {
+            "Phone": ["iPhone 15 Pro Max", "iPhone 15 Pro", "iPhone 15", "iPhone 14 Pro Max", "iPhone 14 Pro", "iPhone 14", "iPhone 13 Pro Max", "iPhone 13 Pro", "iPhone 13", "iPhone 12 Pro Max", "iPhone 12 Pro", "iPhone 12", "iPhone 11 Pro Max", "iPhone 11 Pro", "iPhone 11", "iPhone XS Max", "iPhone XS", "iPhone XR", "iPhone X", "iPhone 8 Plus", "iPhone 8"],
+            "Laptop": ["MacBook Pro 16\" M3 Max", "MacBook Pro 14\" M3 Pro", "MacBook Pro 16\" M2 Max", "MacBook Pro 14\" M2 Pro", "MacBook Air M3", "MacBook Air M2", "MacBook Air M1", "MacBook Pro 13\" M1"]
+        },
+        "Samsung": {
+            "Phone": ["Galaxy S24 Ultra", "Galaxy S24+", "Galaxy S24", "Galaxy S23 Ultra", "Galaxy S23+", "Galaxy S23", "Galaxy S22 Ultra", "Galaxy S22+", "Galaxy S22", "Galaxy Z Fold 5", "Galaxy Z Flip 5", "Galaxy Z Fold 4", "Galaxy Z Flip 4", "Galaxy A54", "Galaxy A34"],
+            "Laptop": ["Galaxy Book4 Pro", "Galaxy Book3 Ultra", "Galaxy Book3 Pro 360", "Galaxy Book2 Pro", "Galaxy Book2"]
+        },
+        "OnePlus": {
+            "Phone": ["OnePlus 12", "OnePlus 11", "OnePlus 10 Pro", "OnePlus 9 Pro", "OnePlus 9", "OnePlus 8T", "OnePlus Nord 3", "OnePlus Nord 2"],
+            "Laptop": []
             },
-            "OnePlus": {
-                "Phone": ["OnePlus 12", "OnePlus 11", "OnePlus 10 Pro", "OnePlus 9 Pro", "OnePlus 9", "OnePlus 8T", "OnePlus Nord 3", "OnePlus Nord 2"],
-                "Laptop": []
-            },
-            "Dell": {
-                "Phone": [],
-                "Laptop": ["XPS 15", "XPS 13", "XPS 17", "Inspiron 15", "Inspiron 14", "Latitude 5430", "Latitude 7430", "Alienware m15", "Alienware x15", "Vostro 15"]
-            },
-            "HP": {
-                "Phone": [],
-                "Laptop": ["Pavilion 15", "Pavilion 14", "Envy 15", "Envy 13", "Spectre x360 14", "Spectre x360 16", "EliteBook 840", "EliteBook 850", "ProBook 450", "Omen 15"]
-            },
-            "Lenovo": {
-                "Phone": [],
-                "Laptop": ["ThinkPad X1 Carbon", "ThinkPad X1 Yoga", "ThinkPad T14", "ThinkPad T16", "IdeaPad Slim 5", "IdeaPad Gaming 3", "Yoga 9i", "Yoga 7i", "Legion 5 Pro"]
-            },
-            "Other": {
-                "Phone": ["Other Model"],
-                "Laptop": ["Other Model"]
-            }
+        "Dell": {
+            "Phone": [],
+            "Laptop": ["XPS 15", "XPS 13", "XPS 17", "Inspiron 15", "Inspiron 14", "Latitude 5430", "Latitude 7430", "Alienware m15", "Alienware x15", "Vostro 15"]
+        },
+        "HP": {
+            "Phone": [],
+            "Laptop": ["Pavilion 15", "Pavilion 14", "Envy 15", "Envy 13", "Spectre x360 14", "Spectre x360 16", "EliteBook 840", "EliteBook 850", "ProBook 450", "Omen 15"]
+        },
+        "Lenovo": {
+            "Phone": [],
+            "Laptop": ["ThinkPad X1 Carbon", "ThinkPad X1 Yoga", "ThinkPad T14", "ThinkPad T16", "IdeaPad Slim 5", "IdeaPad Gaming 3", "Yoga 9i", "Yoga 7i", "Legion 5 Pro"]
+        },
+        "Other": {
+            "Phone": ["Other Model"],
+            "Laptop": ["Other Model"]
         }
+    }
         
-        available_models = model_options.get(brand, {}).get(device_type, ["Other Model"])
+    available_models = model_options.get(brand, {}).get(device_type, ["Other Model"])
         
-        with col_b:
-            if available_models:
-                model = st.selectbox("Model", available_models)
-            else:
-                st.warning(f"No {device_type} models available for {brand}")
-                model = st.text_input("Model (enter manually)", "")
+    with col_b:
+        if available_models:
+            model = st.selectbox("Model", available_models)
+        else:
+            st.warning(f"No {device_type} models available for {brand}")
+            model = st.text_input("Model (enter manually)", "")
             
-            original_price = st.number_input("Original Price (₹)", min_value=0, value=79900, step=1000)
+        original_price = st.number_input("Original Price (₹)", min_value=0, value=79900, step=1000)
         
-        age_months = st.slider("Age (months)", 0, 60, 18, help="How old is the device?")
-        if st.session_state.processing:
-           st.warning("⏳ Processing… please wait")
+    age_months = st.slider("Age (months)", 0, 60, 18, help="How old is the device?")
+    if st.session_state.processing:
+        st.warning("⏳ Processing… please wait")
         # Form submit button
+    with st.form("diagnosis_form"):
         submitted = st.form_submit_button(" Diagnose Device", type="primary")
 
 with col2: