Update app.py

app.py

@@ -11,7 +11,6 @@ import logging
 from datetime import datetime
 import hashlib
 import shutil
-import requests
 import gradio as gr
 from google import genai
 from google.genai import types
@@ -36,26 +35,6 @@ logger = logging.getLogger(__name__)
 LOGS_DIR = "generation_logs"
 os.makedirs(LOGS_DIR, exist_ok=True)
 
-def get_location_from_ip(ip):
-    """Get approximate location from IP address"""
-    if ip == "unknown" or ip.startswith("127.") or ip.startswith("192.168."):
-        return {"country": "unknown", "city": "unknown"}
-    
-    try:
-        response = requests.get(f"https://ipapi.co/{ip}/json/", timeout=2)
-        data = response.json()
-        return {
-            "ip": ip,
-            "country": data.get("country_name", "unknown"),
-            "city": data.get("city", "unknown"),
-            "region": data.get("region", "unknown"),
-            "latitude": data.get("latitude"),
-            "longitude": data.get("longitude")
-        }
-    except Exception as e:
-        logger.error(f"Geolocation failed: {e}")
-        return {"country": "unknown", "city": "unknown"}
-
 def save_generation_log(prompt, images_dict, metadata=None, request_info=None):
     """Save generation logs with images and user info"""
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -70,7 +49,6 @@ def save_generation_log(prompt, images_dict, metadata=None, request_info=None):
         "session_id": session_id,
         "prompt": prompt,
         "client_ip": request_info.get("ip") if request_info else None,
-        "location": request_info.get("location") if request_info else None,
         "user_agent": request_info.get("user_agent") if request_info else None,
         "metadata": metadata or {}
     }
@@ -360,14 +338,14 @@ def edit_3d_model(edit_prompt: str, transparent_image_path: str) -> str:
 def generate_3d_ui(prompt, request: gr.Request, progress=gr.Progress()):
     """UI wrapper with progress updates"""
     
+    # Get client info
     client_ip = request.headers.get("x-forwarded-for", "").split(",")[0].strip()
     if not client_ip:
         client_ip = request.client.host if request else "unknown"
-    location = get_location_from_ip(client_ip)
     user_agent = request.headers.get("user-agent", "unknown") if request else "unknown"
     
     logger.info(f"=== NEW GENERATION REQUEST ===")
-    logger.info(f"IP: {client_ip} | Location: {location.get('city', 'unknown')}, {location.get('country', 'unknown')}")
+    logger.info(f"IP: {client_ip}")
     logger.info(f"Prompt: {prompt}")
     
     if not client:
@@ -476,7 +454,6 @@ def generate_3d_ui(prompt, request: gr.Request, progress=gr.Progress()):
     }
     request_info = {
         "ip": client_ip,
-        "location": location,
         "user_agent": user_agent
     }
     save_generation_log(prompt, images_dict, metadata, request_info)
@@ -500,12 +477,13 @@ def edit_3d_ui(edit_prompt, current_image, edit_count, request: gr.Request, prog
     """UI wrapper for editing"""
     
     # Get client info
-    client_ip = request.client.host if request else "unknown"
-    location = get_location_from_ip(client_ip)
+    client_ip = request.headers.get("x-forwarded-for", "").split(",")[0].strip()
+    if not client_ip:
+        client_ip = request.client.host if request else "unknown"
     user_agent = request.headers.get("user-agent", "unknown") if request else "unknown"
     
     logger.info(f"=== EDIT REQUEST #{edit_count + 1} ===")
-    logger.info(f"IP: {client_ip} | Location: {location.get('city', 'unknown')}, {location.get('country', 'unknown')}")
+    logger.info(f"IP: {client_ip}")
     logger.info(f"Edit: {edit_prompt}")
     
     if current_image is None:
@@ -591,7 +569,6 @@ def edit_3d_ui(edit_prompt, current_image, edit_count, request: gr.Request, prog
     }
     request_info = {
         "ip": client_ip,
-        "location": location,
         "user_agent": user_agent
     }
     save_generation_log(edit_prompt, images_dict, metadata, request_info)
@@ -653,7 +630,7 @@ with gr.Blocks() as main_ui:
     
     ⏱️ *Generation takes 1-2 minutes. First run may take longer as the model warms up.*
     
-    📍 *Note: IP addresses and approximate locations are logged for analytics.*
+    📍 *Note: IP addresses are logged for analytics.*
     """)
     
     gr.Markdown("## 1️⃣ Generate Initial 3D Model")