Update app.py

app.py

@@ -38,24 +38,30 @@ def is_port_available(port):
 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}")
+    port = PORT
+    max_attempts = 5
+    for i in range(max_attempts):
+        if is_port_available(port):
+            try:
+                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}")
+                return port
+            except Exception as e:
+                print(f"Failed to start HTTP server on port {port}: {e}")
+                raise
         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
+            print(f"Port {port} is in use, trying port {port + 1}")
+            port += 1
+    raise OSError("No available ports found after multiple attempts")
 
 # Start ngrok to create a public URL
-def start_ngrok():
+def start_ngrok(port):
     global ngrok_process
     try:
-        ngrok_process = subprocess.Popen(["ngrok", "http", str(PORT)], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        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"]
@@ -65,12 +71,12 @@ def start_ngrok():
         print(f"Failed to start ngrok: {e}")
         raise
 
-# 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
-)
+# Hugging Face captioning pipeline
+try:
+    caption_pipeline = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+except Exception as e:
+    print(f"Failed to initialize Hugging Face pipeline: {e}")
+    caption_pipeline = None
 
 @app.route('/')
 def index():
@@ -78,6 +84,7 @@ def index():
 
 @app.route('/upload', methods=['POST'])
 def upload_image():
+    global ngrok_process
     try:
         # Get base64 image from request
         image_data = request.json['image'].split(',')[1]
@@ -89,12 +96,18 @@ def upload_image():
             f.write(image_bytes)
         
         # Generate caption using Hugging Face
-        image = Image.open(io.BytesIO(image_bytes))
-        caption_result = caption_pipeline(image)
-        caption = caption_result[0]['generated_text']
+        caption = "Default caption: Uploaded image"
+        if caption_pipeline:
+            try:
+                image = Image.open(io.BytesIO(image_bytes))
+                caption_result = caption_pipeline(image)
+                caption = caption_result[0]['generated_text']
+            except Exception as e:
+                print(f"Failed to generate caption: {e}")
 
         # Get public URL from ngrok
-        public_url = start_ngrok()
+        port = start_http_server()
+        public_url = start_ngrok(port)
         image_url = f"{public_url}/uploads/captured_image.jpg"
 
         # Create Instagram media container
@@ -149,14 +162,4 @@ def shutdown():
         return jsonify({"status": "error", "message": str(e)})
 
 if __name__ == '__main__':
-    try:
-        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:
-        if httpd:
-            httpd.shutdown()
-            httpd.server_close()
-        if ngrok_process:
-            ngrok_process.terminate()
+    app.run(debug=True)