11

app.py

@@ -1,3 +1,4 @@
+import socket
 from fastapi import FastAPI, HTTPException, UploadFile, File, Form
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse
@@ -408,6 +409,24 @@ def get_memory_usage():
 async def memory_status():
     return get_memory_usage()
 
+@app.get("/ip")
+async def get_ip():
+    # Get internal IP
+    hostname = socket.gethostname()
+    internal_ip = socket.gethostbyname(hostname)
+    
+    # Get external IP
+    try:
+        external_ip = requests.get('https://api.ipify.org').text
+    except:
+        external_ip = "Unable to fetch external IP"
+    
+    return {
+        "hostname": hostname,
+        "internal_ip": internal_ip,
+        "external_ip": external_ip,
+        "timestamp": datetime.datetime.now().isoformat()
+    }
 # Add new endpoint
 @app.get("/proxy")
 async def proxy_request(url: str, request: Request):