Revert to original GeoTIFF-only implementation

- Removed multi-format image support (PNG, JPG, etc.)
- Restored original process_geotiff_file function
- Restored GeoTIFF-only file input
- Restored original advanced_extraction.py implementation
- Back to simple, working GeoTIFF tree detection

app.py

@@ -154,59 +154,44 @@ def create_split_view_map(geojson_data, bounds):
         m = folium.Map(location=[40.7, -74.0], zoom_start=10)
         return m
 
-def process_image_file(image_file):
-    """Process uploaded image file for tree detection."""
-    if image_file is None:
-        return None, "Please upload an image file or use the example file"
+def process_geotiff_file(geotiff_file):
+    """Process uploaded GeoTIFF file for tree detection."""
+    if geotiff_file is None:
+        return None, "Please upload a GeoTIFF file or use the example file"
 
     try:
         # Create unique ID
         unique_id = str(uuid.uuid4().hex)[:8]
         
         # Handle file upload
-        if hasattr(image_file, 'name'):
-            filename = os.path.basename(image_file.name)
+        if hasattr(geotiff_file, 'name'):
+            filename = os.path.basename(geotiff_file.name)
         else:
-            filename = os.path.basename(image_file)
+            filename = os.path.basename(geotiff_file)
 
         # Save uploaded file
-        image_path = os.path.join(TEMP_DIRS['uploads'], f"{unique_id}_{filename}")
+        geotiff_path = os.path.join(TEMP_DIRS['uploads'], f"{unique_id}_{filename}")
 
-        if hasattr(image_file, 'read'):
-            file_content = image_file.read()
-            with open(image_path, "wb") as f:
+        if hasattr(geotiff_file, 'read'):
+            file_content = geotiff_file.read()
+            with open(geotiff_path, "wb") as f:
                 f.write(file_content)
         else:
-            shutil.copy(image_file, image_path)
+            shutil.copy(geotiff_file, geotiff_path)
 
-        logger.info(f"File saved to {image_path}")
+        logger.info(f"File saved to {geotiff_path}")
 
-        # Check if it's a GeoTIFF file for advanced processing
-        if filename.lower().endswith(('.tif', '.tiff')):
-            # Use advanced extraction for GeoTIFF files
-            from utils.advanced_extraction import extract_features_from_geotiff
-            
-            logger.info("Extracting tree features from GeoTIFF...")
-            geojson_data = extract_features_from_geotiff(image_path, TEMP_DIRS['processed'], "trees")
-        else:
-            # Use general image processing for other formats
-            from utils.geospatial import process_image_to_geojson
-            from utils.image_processing import process_image
-            
-            logger.info("Processing regular image for tree detection...")
-            processed_image_path = process_image(image_path, TEMP_DIRS['processed'])
-            geojson_data = process_image_to_geojson(processed_image_path, feature_type="trees", original_file_path=image_path)
+        # Import and extract features
+        from utils.advanced_extraction import extract_features_from_geotiff
+        
+        logger.info("Extracting tree features...")
+        geojson_data = extract_features_from_geotiff(geotiff_path, TEMP_DIRS['processed'], "trees")
 
         if not geojson_data or not geojson_data.get('features'):
             return None, "No trees detected in the image"
 
         # Get bounds and create map
-        if filename.lower().endswith(('.tif', '.tiff')):
-            bounds = get_bounds_from_geotiff(image_path)
-        else:
-            # For regular images, use default bounds or extract from metadata
-            bounds = get_bounds_from_geotiff(image_path)  # This will use defaults for non-GeoTIFF files
-        
+        bounds = get_bounds_from_geotiff(geotiff_path)
         map_obj = create_split_view_map(geojson_data, bounds)
 
         if map_obj:
@@ -304,7 +289,7 @@ def create_gradio_interface():
         gr.HTML("""
         <div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 10px; margin-bottom: 20px;">
             <h1 style="color: white; margin: 0; font-size: 2.5em;">🌲 ForestAI</h1>
-            <p style="color: white; margin: 10px 0 0 0; font-size: 1.2em;">Tree Detection from Satellite & Aerial Imagery</p>
+            <p style="color: white; margin: 10px 0 0 0; font-size: 1.2em;">Tree Detection from Satellite Imagery</p>
         </div>
         """)
 
@@ -313,8 +298,8 @@ def create_gradio_interface():
                 gr.Markdown("### Upload GeoTIFF File")
                 
                 file_input = gr.File(
-                    label="Select Image File",
-                    file_types=[".tif", ".tiff", ".png", ".jpg", ".jpeg", ".bmp", ".gif"],
+                    label="Select GeoTIFF File",
+                    file_types=[".tif", ".tiff"],
                     type="filepath"
                 )
                 
@@ -359,7 +344,7 @@ def create_gradio_interface():
                                 display:flex; align-items:center; justify-content:center; 
                                 background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);">
                         <div style="text-align:center; color:#666;">
-                            <h3>🌲 Upload an image file or use example to see detected trees</h3>
+                            <h3>🌲 Upload a GeoTIFF file or use example to see detected trees</h3>
                             <p>Interactive map will appear here</p>
                         </div>
                     </div>
@@ -369,7 +354,7 @@ def create_gradio_interface():
 
         # Event handlers
         analyze_btn.click(
-            fn=process_image_file,
+            fn=process_geotiff_file,
             inputs=[file_input],
             outputs=[map_output, status_output],
             show_progress=True
@@ -385,16 +370,11 @@ def create_gradio_interface():
         # Simple instructions
         gr.Markdown("""
         ### How to Use:
-        1. **Upload** an image file (GeoTIFF, PNG, JPG, etc.) OR click "Use Example File" to try with the included sample
+        1. **Upload** a GeoTIFF satellite image file OR click "Use Example File" to try with the included sample
         2. **Click** "Detect Trees" to analyze your uploaded image
         3. **Explore** the interactive map with detected tree areas
         4. **Use** the split-view slider to compare base map and satellite imagery
 
-        ### Supported Formats:
-        - **GeoTIFF (.tif, .tiff)**: Best for satellite imagery with geographic data
-        - **Regular Images (.png, .jpg, .jpeg, .bmp, .gif)**: For general image analysis
-        - **Processing**: GeoTIFF files use advanced NDVI analysis, other formats use general image processing
-
         ### Map Controls:
         - **Split View**: Drag the vertical slider to compare layers
         - **Zoom**: Scroll to zoom in/out, drag to pan

utils/advanced_extraction.py

@@ -83,4 +83,4 @@ def extract_features_from_geotiff(image_path, output_folder, feature_type="trees
         
     except Exception as e:
         logging.error(f"Error extracting features: {str(e)}")
-        return {"type": "FeatureCollection", "features": []}
+        return {"type": "FeatureCollection", "features": []}