Update app.py

app.py

@@ -1,22 +1,19 @@
 import gradio as gr
-import os
 from classifier import analyze_image
 
 def analyze(image):
     if image is None:
         return "Please upload an image."
 
-    # Save image temporarily
-    image_path = "temp.jpg"
-    image.save(image_path)
+    image.save("temp.jpg")
+    result = analyze_image("temp.jpg")
 
-    result = analyze_image(image_path)
+    return f"""
+🧠 **Detected Object:** {result['detected_object']}  
+📊 **Object Confidence:** {result['object_confidence']}
 
-    output = f"""
-🧠 **Detected Object:** {result['detected_object']}
-📊 **Confidence:** {result['confidence']}
-
-♻️ **Waste Category:** {result['waste_category']}
+♻️ **Waste Category:** {result['waste_category']}  
+📊 **Waste Confidence:** {result['waste_confidence']}
 
 📝 **Reason:**  
 {result['reason']}
@@ -27,35 +24,23 @@ def analyze(image):
 🌍 **Environmental Impact:**  
 {result['environmental_impact']}
 """
-    return output
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        """
-        # ♻️ Smart Waste AI  
-        ### AI-powered Waste Segregation Advisor  
-        Upload an image of waste and get instant guidance.
-        """
-    )
-
-    with gr.Row():
-        image_input = gr.Image(type="pil", label="Upload Waste Image")
+    gr.Markdown("""
+    # ♻️ Smart Waste AI  
+    **Hybrid AI-powered Waste Segregation Advisor**  
+    ResNet + CLIP (Zero-shot Vision Reasoning)
+    """)
+
+    image = gr.Image(type="pil", label="Upload Waste Image")
+    button = gr.Button("Analyze Waste", variant="primary")
+    output = gr.Markdown()
+
+    button.click(analyze, image, output)
+
+    gr.Markdown("""
+    ---
+    🔗 **GitHub:** https://github.com/OGrohit/smart-waste-ai  
+    """)
     
-    analyze_btn = gr.Button("Analyze Waste", variant="primary")
-    output_box = gr.Markdown()
-
-    analyze_btn.click(
-        fn=analyze,
-        inputs=image_input,
-        outputs=output_box
-    )
-
-    gr.Markdown(
-        """
-        ---
-        🔗 **GitHub Repository:**  
-        https://github.com/rohitdecodes/smart-waste-ai
-        """
-    )
-
 demo.launch()