Update app.py

app.py

@@ -2,9 +2,18 @@ import gradio as gr
 import numpy as np
 from part1_data import TobaccoAnalyzer
 from part2_visualization import VisualizationHandler
+from part3 import SAMAnalyzer
+
+if not os.path.exists('sam_vit_h_4b8939.pth'):
+    print("Downloading SAM model...")
+    urllib.request.urlretrieve(
+        'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth',
+        'sam_vit_h_4b8939.pth'
+    )
+    print("SAM model downloaded successfully")
 
 def analyze_location(location_name):
-    """Main analysis function with enhanced historical and forecast data"""
+    """Analyze location using weather and satellite data"""
     try:
         analyzer = TobaccoAnalyzer()
         visualizer = VisualizationHandler(analyzer.optimal_conditions)
@@ -39,7 +48,7 @@ def analyze_location(location_name):
         if trends is None:
             return None, None, "Error calculating trends.", None
         
-        # Calculate overall score with weights
+        # Calculate overall score
         weights = {
             'temperature': 0.3,
             'humidity': 0.2,
@@ -59,7 +68,7 @@ def analyze_location(location_name):
         gauge_plot = visualizer.create_gauge_chart(overall_score)
         location_map = visualizer.create_enhanced_map(lat, lon, overall_score, historical['estimated_ndvi'].mean())
         
-        # Generate comprehensive analysis text
+        # Generate analysis text
         analysis_text = f"""
         📍 Location Analysis:
         Location: {location_data['address']}
@@ -110,6 +119,21 @@ def analyze_location(location_name):
         print(f"Error details: {e}")
         return None, None, error_message, None
 
+def analyze_satellite_image(image):
+    """Analyze satellite image using SAM2"""
+    try:
+        analyzer = SAMAnalyzer()
+        veg_index, health_analysis, viz_plot = analyzer.process_image(image)
+        
+        if veg_index is None:
+            return None, "Error processing image. Please try again."
+        
+        analysis_text = analyzer.format_analysis_text(health_analysis)
+        
+        return viz_plot, analysis_text
+    except Exception as e:
+        return None, f"Error analyzing image: {str(e)}"
+
 def get_vegetation_status(ndvi):
     """Get detailed vegetation status based on NDVI value"""
     if ndvi < 0:
@@ -166,19 +190,14 @@ def get_risk_factors(df, trends):
     """Analyze and return risk factors"""
     risks = []
     
-    # Temperature risks
     if df['temp_range'].mean() > 15:
         risks.append("• High daily temperature variations")
     if trends['historical']['temperature']['trend'] < 0:
         risks.append("• Declining temperature trend")
-    
-    # Rainfall risks
     if df['rainfall'].std() > df['rainfall'].mean():
         risks.append("• Inconsistent rainfall patterns")
     if trends['historical']['rainfall']['trend'] < 0:
         risks.append("• Decreasing rainfall trend")
-    
-    # NDVI risks
     if trends['historical']['ndvi']['trend'] < 0:
         risks.append("• Declining vegetation health")
     
@@ -214,62 +233,96 @@ def get_long_term_outlook(trends):
         return "Deteriorating conditions"
     return "Mixed conditions - monitor closely"
 
-# Create Gradio interface with enhanced layout
+# Create Gradio interface
 with gr.Blocks(theme=gr.themes.Base()) as demo:
     gr.Markdown(
         """
         # 🌱 Agricultural Credit Risk Assessment System
-        Comprehensive Weather, Vegetation, and Credit Scoring Analysis for Tobacco Farming
+        ## Weather, Vegetation, and Credit Scoring Analysis for Tobacco Farming
         """
     )
     
-    # Input Section
-    with gr.Row():
-        location_input = gr.Textbox(
-            label="Enter Location",
-            placeholder="e.g., Tabora, Tanzania",
-            scale=4
-        )
-        analyze_button = gr.Button("Analyze", variant="primary", scale=1)
-
-    # Map and Analysis Section
-    with gr.Row():
-        with gr.Column(scale=2):  # Larger map
-            location_map = gr.HTML(label="NDVI Analysis Map")
-        with gr.Column(scale=1):
-            analysis_text = gr.Textbox(
-                label="Analysis Results",
-                lines=25,
-                show_label=False
+    with gr.Tab("📊 Location Analysis"):
+        # Input Section
+        with gr.Row():
+            location_input = gr.Textbox(
+                label="Enter Location",
+                placeholder="e.g., Tabora, Tanzania",
+                scale=4
             )
+            analyze_button = gr.Button("Analyze", variant="primary", scale=1)
+
+        # Map and Analysis Section
+        with gr.Row():
+            with gr.Column(scale=2):
+                location_map = gr.HTML(label="NDVI Analysis Map")
+            with gr.Column(scale=1):
+                analysis_text = gr.Textbox(
+                    label="Analysis Results",
+                    lines=25,
+                    show_label=False
+                )
+        
+        # Plots Section
+        with gr.Row():
+            weather_plot = gr.Plot(label="Weather Analysis")
+        
+        with gr.Row():
+            score_gauge = gr.Plot(label="Growing Conditions Score")
+        
+        # Location Examples
+        gr.Examples(
+            examples=[
+                ["Tabora, Tanzania"],
+                ["Urambo, Tabora, Tanzania"],
+                ["Sikonge, Tabora, Tanzania"],
+                ["Nzega, Tabora, Tanzania"]
+            ],
+            inputs=location_input,
+            outputs=[location_map, analysis_text, weather_plot, score_gauge],
+            fn=analyze_location,
+            cache_examples=True
+        )
+        
+        # Handle location analysis
+        analyze_button.click(
+            fn=analyze_location,
+            inputs=[location_input],
+            outputs=[location_map, analysis_text, weather_plot, score_gauge]
+        )
     
-    # Plots Section
-    with gr.Row():
-        weather_plot = gr.Plot(label="Weather Analysis")
-    
-    with gr.Row():
-        score_gauge = gr.Plot(label="Growing Conditions Score")
-    
-    # Examples
-    gr.Examples(
-        examples=[
-            ["Tabora, Tanzania"],
-            ["Urambo, Tabora, Tanzania"],
-            ["Sikonge, Tabora, Tanzania"],
-            ["Nzega, Tabora, Tanzania"]
-        ],
-        inputs=location_input,
-        outputs=[location_map, analysis_text, weather_plot, score_gauge],
-        fn=analyze_location,
-        cache_examples=True
-    )
-    
-    # Set up the click event
-    analyze_button.click(
-        fn=analyze_location,
-        inputs=[location_input],
-        outputs=[location_map, analysis_text, weather_plot, score_gauge]
-    )
+    with gr.Tab("🛰️ Satellite Image Analysis"):
+        gr.Markdown("""
+        ## Satellite Image Analysis with SAM2
+        Upload a satellite or aerial image to analyze vegetation health using advanced segmentation.
+        """)
+        
+        with gr.Row():
+            image_input = gr.Image(
+                label="Upload Satellite/Aerial Image",
+                type="file",
+                tool="select"
+            )
+        
+        with gr.Row():
+            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
+                )
+        
+        # Handle image analysis
+        analyze_image_button.click(
+            fn=analyze_satellite_image,
+            inputs=[image_input],
+            outputs=[image_plot, image_analysis]
+        )
 
 # Launch the app
 if __name__ == "__main__":