Update app.py

app.py

@@ -148,7 +148,7 @@ def process_and_analyze_image(image):
         analyzer = SAMAnalyzer(model_path=MODEL_PATH)
         
         print("Processing image...")
-        veg_index, health_analysis, viz_plot = analyzer.process_image(image)
+        veg_index, health_analysis, fig = analyzer.process_image(image)
         
         if veg_index is None:
             return None, "Error processing image.", "Image processing failed"
@@ -164,12 +164,10 @@ def process_and_analyze_image(image):
         • High Vegetation: {health_analysis['health_distribution']['high_vegetation']*100:.1f}%
         
         📋 Overall Health Status: {health_analysis['overall_health']}
-        
-        Note: Analysis uses SAM2 for farmland segmentation
         """
         
         debug_msg = "Analysis completed successfully"
-        return viz_plot, analysis_text, debug_msg
+        return fig, analysis_text, debug_msg
         
     except Exception as e:
         error_msg = f"Error during analysis: {str(e)}"
@@ -344,14 +342,15 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
             analyze_image_button = gr.Button("🔍 Analyze Image", variant="primary")
         
         with gr.Row():
-            with gr.Column():
-                image_plot = gr.Plot(label="Vegetation Analysis Results")
-            with gr.Column():
-                image_analysis = gr.Textbox(
-                    label="Analysis Results",
-                    lines=10,
-                    show_label=False
-                )
+            image_plot = gr.Plot(
+                label="Vegetation Analysis Results"
+            )
+            
+        with gr.Row():
+            image_analysis = gr.Textbox(
+                label="Analysis Results",
+                lines=10
+            )
         
         debug_output = gr.Textbox(
             label="Debug Information",