Update app.py

app.py

@@ -347,20 +347,18 @@ def extract_info(template, text):
         print(f"Error in extract_info: {e}\n{trace}")
         return f"❌ Fehler: {str(e)}", "{}"
 @spaces.GPU
+@spaces.GPU
 def create_map(df, location_col):
-    # Create a log file to track what's happening
-    with open("map_debug.log", "w") as log_file:
-        log_file.write(f"Starting map creation at {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
-        log_file.write(f"Data shape: {df.shape}, Location column: {location_col}\n")
+    # Start a simple log to track execution
+    with open("map_debug.log", "w") as log:
+        log.write(f"Starting map creation at {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
     
-    # Create the map
     m = folium.Map(
         location=[20, 0],
         zoom_start=2, 
         control_scale=True
     )
     
-    # Add base map and controls
     folium.TileLayer(
         tiles=MAP_TILES["GreenMap"]["url"],
         attr=MAP_TILES["GreenMap"]["attr"],
@@ -372,78 +370,70 @@ def create_map(df, location_col):
     Fullscreen().add_to(m)
     MeasureControl(position='topright', primary_length_unit='kilometers').add_to(m)
     
-    # Setup for geocoding and markers
     geocoder = SafeGeocoder()
     coords = []
     marker_cluster = MarkerCluster(name="Locations").add_to(m)
     processed_count = 0
     
-    # Log marker creation process
-    with open("map_debug.log", "a") as log_file:
-        log_file.write("\nStarting to process locations...\n")
+    with open("map_debug.log", "a") as log:
+        log.write(f"Processing {len(df)} rows from dataframe\n")
     
-    # Process each row in dataframe
     for idx, row in df.iterrows():
-        # Log the current row we're processing
-        with open("map_debug.log", "a") as log_file:
-            log_file.write(f"\nProcessing row {idx}\n")
-        
         if pd.isna(row[location_col]):
-            with open("map_debug.log", "a") as log_file:
-                log_file.write(f"  - Empty location, skipping\n")
             continue
             
         location = str(row[location_col]).strip()
-        with open("map_debug.log", "a") as log_file:
-            log_file.write(f"  - Location: {location}\n")
         
-        # Build additional info string for the popup
+        # Log the location being processed
+        with open("map_debug.log", "a") as log:
+            log.write(f"Processing location: {location}\n")
+        
         additional_info = ""
         for col in df.columns:
             if col != location_col and not pd.isna(row[col]):
                 additional_info += f"<br><b>{col}:</b> {row[col]}"
         
-        # Split location if it contains multiple comma-separated places
         try:
             locations = [loc.strip() for loc in location.split(',') if loc.strip()]
             if not locations:
                 locations = [location]
         except Exception as e:
-            with open("map_debug.log", "a") as log_file:
-                log_file.write(f"  - Error splitting location: {str(e)}\n")
+            with open("map_debug.log", "a") as log:
+                log.write(f"Error splitting location '{location}': {str(e)}\n")
             locations = [location]
+        
+        # Log the parsed locations
+        with open("map_debug.log", "a") as log:
+            log.write(f"Split into locations: {locations}\n")
             
-        with open("map_debug.log", "a") as log_file:
-            log_file.write(f"  - Parsed locations: {locations}\n")
-            
-        # Process each individual location
         for loc in locations:
-            with open("map_debug.log", "a") as log_file:
-                log_file.write(f"    - Processing: {loc}\n")
+            try:
+                # Log the current location
+                with open("map_debug.log", "a") as log:
+                    log.write(f"Getting coordinates for: {loc}\n")
                 
-            # Get coordinates from geocoder
-            point = geocoder.get_coords(loc)
-            
-            with open("map_debug.log", "a") as log_file:
-                log_file.write(f"    - Coordinates: {point}\n")
+                point = geocoder.get_coords(loc)
                 
-            if point:
-                # Create popup content - this is the part that was disappearing
-                popup_content = f"""
-                <div style="min-width: 200px; max-width: 300px">
-                    <h4 style="font-family: 'Source Sans Pro', sans-serif; margin-bottom: 5px;">{loc}</h4>
-                    <div style="font-family: 'Source Sans Pro', sans-serif; font-size: 14px;">
-                        {additional_info}
+                if point:
+                    # Log the found coordinates
+                    with open("map_debug.log", "a") as log:
+                        log.write(f"Found coordinates for {loc}: {point}\n")
+                    
+                    # Create popup HTML - this is the part that was disappearing
+                    popup_content = f"""
+                    <div style="min-width: 200px; max-width: 300px">
+                        <h4 style="font-family: 'Source Sans Pro', sans-serif; margin-bottom: 5px;">{loc}</h4>
+                        <div style="font-family: 'Source Sans Pro', sans-serif; font-size: 14px;">
+                            {additional_info}
+                        </div>
                     </div>
-                </div>
-                """
-                
-                # Log the popup content to see exactly what's being created
-                with open("map_debug.log", "a") as log_file:
-                    log_file.write(f"    - Created popup content for {loc}\n")
-                
-                # Add the marker to the map
-                try:
+                    """
+                    
+                    # Log popup creation
+                    with open("map_debug.log", "a") as log:
+                        log.write(f"Created popup for {loc}\n")
+                    
+                    # Add marker with popup
                     folium.Marker(
                         location=point,
                         popup=folium.Popup(popup_content, max_width=300),
@@ -451,25 +441,27 @@ def create_map(df, location_col):
                         icon=folium.Icon(color="blue", icon="info-sign")
                     ).add_to(marker_cluster)
                     
+                    # Log marker addition
+                    with open("map_debug.log", "a") as log:
+                        log.write(f"Added marker for {loc}\n")
+                    
                     coords.append(point)
                     processed_count += 1
-                    
-                    with open("map_debug.log", "a") as log_file:
-                        log_file.write(f"    - Added marker for {loc}\n")
-                except Exception as e:
-                    with open("map_debug.log", "a") as log_file:
-                        log_file.write(f"    - Error adding marker: {str(e)}\n")
+                else:
+                    # Log geocoding failure
+                    with open("map_debug.log", "a") as log:
+                        log.write(f"No coordinates found for {loc}\n")
+            except Exception as e:
+                # Log any errors in processing this location
+                with open("map_debug.log", "a") as log:
+                    log.write(f"Error processing {loc}: {str(e)}\n")
+                    log.write(traceback.format_exc() + "\n")
     
     # Fit map bounds if we have coordinates
     if coords:
         m.fit_bounds(coords)
     
-    # Log completion information
-    with open("map_debug.log", "a") as log_file:
-        log_file.write(f"\nMap creation completed at {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
-        log_file.write(f"Processed {processed_count} locations\n")
-    
-    # Add custom CSS for better display
+    # Custom CSS for map
     custom_css = """
     <style>
     @import url('https://fonts.googleapis.com/css2?family=Source+Sans+Pro:wght@400;600&display=swap');
@@ -487,6 +479,11 @@ def create_map(df, location_col):
     """
     m.get_root().header.add_child(folium.Element(custom_css))
     
+    # Log completion
+    with open("map_debug.log", "a") as log:
+        log.write(f"Map creation completed at {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
+        log.write(f"Processed {processed_count} locations\n")
+    
     return m._repr_html_(), processed_count
     
     custom_css = """