Update app.py

app.py

@@ -7,6 +7,7 @@ import socketserver
 import threading
 import subprocess
 import time
+import socket
 from dotenv import load_dotenv
 from PIL import Image
 import io
@@ -22,30 +23,54 @@ INSTAGRAM_USER_ID = os.getenv('INSTAGRAM_USER_ID')
 HUGGINGFACE_API_TOKEN = os.getenv('HUGGINGFACE_API_TOKEN')
 UPLOAD_FOLDER = 'uploads'
 PORT = 8000
+httpd = None
+ngrok_process = None
 
 # Ensure upload folder exists
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
+# Check if port is available
+def is_port_available(port):
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        return s.connect_ex(('localhost', port)) != 0
+
 # Start local HTTP server to serve images
-Handler = http.server.SimpleHTTPRequestHandler
-httpd = socketserver.TCPServer(("", PORT), Handler)
-server_thread = threading.Thread(target=httpd.serve_forever)
-server_thread.daemon = True
-server_thread.start()
+def start_http_server():
+    global httpd
+    Handler = http.server.SimpleHTTPRequestHandler
+    try:
+        if is_port_available(PORT):
+            httpd = socketserver.TCPServer(("", PORT), Handler)
+            server_thread = threading.Thread(target=httpd.serve_forever)
+            server_thread.daemon = True
+            server_thread.start()
+            print(f"HTTP server started on port {PORT}")
+        else:
+            raise OSError(f"Port {PORT} is already in use. Please free the port or choose another.")
+    except Exception as e:
+        print(f"Failed to start HTTP server: {e}")
+        raise
 
 # Start ngrok to create a public URL
-ngrok_process = None
-
 def start_ngrok():
     global ngrok_process
-    ngrok_process = subprocess.Popen(["ngrok", "http", str(PORT)], stdout=subprocess.PIPE)
-    time.sleep(2)  # Wait for ngrok to start
-    resp = requests.get("http://localhost:4040/api/tunnels")
-    public_url = resp.json()["tunnels"][0]["public_url"]
-    return public_url
+    try:
+        ngrok_process = subprocess.Popen(["ngrok", "http", str(PORT)], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        time.sleep(2)  # Wait for ngrok to start
+        resp = requests.get("http://localhost:4040/api/tunnels")
+        public_url = resp.json()["tunnels"][0]["public_url"]
+        print(f"Ngrok tunnel started at {public_url}")
+        return public_url
+    except Exception as e:
+        print(f"Failed to start ngrok: {e}")
+        raise
 
-# Hugging Face captioning pipeline
-caption_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+# Hugging Face captioning pipeline with warning suppression
+caption_pipeline = pipeline(
+    "image-to-text",
+    model="Salesforce/blip-image-captioning-base",
+    clean_up_tokenization_spaces=True  # Suppress FutureWarning
+)
 
 @app.route('/')
 def index():
@@ -105,18 +130,33 @@ def upload_image():
         # Clean up ngrok
         if ngrok_process:
             ngrok_process.terminate()
+            ngrok_process = None
 
 @app.route('/shutdown', methods=['POST'])
 def shutdown():
-    httpd.shutdown()
-    if ngrok_process:
-        ngrok_process.terminate()
-    return jsonify({"status": "success", "message": "Server shutdown"})
+    global httpd, ngrok_process
+    try:
+        if httpd:
+            httpd.shutdown()
+            httpd.server_close()
+            print("HTTP server shut down")
+        if ngrok_process:
+            ngrok_process.terminate()
+            ngrok_process = None
+            print("Ngrok shut down")
+        return jsonify({"status": "success", "message": "Server shutdown"})
+    except Exception as e:
+        return jsonify({"status": "error", "message": str(e)})
 
 if __name__ == '__main__':
     try:
-        app.run(debug=True)
+        start_http_server()
+        app.run(debug=True, use_reloader=False)  # Disable reloader to prevent multiple server instances
+    except Exception as e:
+        print(f"Application failed to start: {e}")
     finally:
-        httpd.shutdown()
+        if httpd:
+            httpd.shutdown()
+            httpd.server_close()
         if ngrok_process:
             ngrok_process.terminate()