Update app.py

app.py

@@ -1,41 +1,36 @@
-import os
-import socket
-import threading
-from http.server import HTTPServer, SimpleHTTPRequestHandler
-import ssl
+# app.py
+import os, socket, threading, sys
 
-print("HI")
-
-def start_https():
-    https_port = int(os.getenv("HTTPS_PORT", "8443"))
-    # create ssl context
-    context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
-    context.load_cert_chain("cert.pem", "key.pem")
-    httpd = HTTPServer(("0.0.0.0", https_port), SimpleHTTPRequestHandler)
-    httpd.socket = context.wrap_socket(httpd.socket, server_side=True)
-    print(f"https listening on 0.0.0.0:{https_port}")
-    httpd.serve_forever()
-
-thread = threading.Thread(target=start_https, daemon=True)
-thread.start()
-
-repo_id = os.getenv("REPO_ID", "spaceslab/thost")
-user, space = repo_id.split("/")
+host = "0.0.0.0"
 port = int(os.getenv("PORT", "5678"))
-host = f"{space}.{user}.hf.space"
 
-print(f"tcp://{host}:{port}")
+print("listening on", host, port)
+print("hostname:", socket.gethostname())
+try:
+    print("local ips:", socket.gethostbyname_ex(socket.gethostname())[2])
+except Exception as e:
+    print("ip lookup error:", e)
+sys.stdout.flush()
 
-sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-sock.bind(("0.0.0.0", port))
-sock.listen()
-print(f"listening on 0.0.0.0:{port}")
-while True:
-    conn, addr = sock.accept()
-    print("connection from", addr)
-    data = conn.recv(1024)
-    if not data:
+def handle_client(conn, addr):
+    print("client connected:", addr)
+    try:
+        while True:
+            data = conn.recv(1024)
+            if not data:
+                break
+            print("received:", data.decode(errors="ignore").strip())
+            conn.sendall(data)
+    except Exception as err:
+        print("error:", err)
+    finally:
         conn.close()
-        continue
-    conn.sendall(data)
-    conn.close()
+        print("client disconnected:", addr)
+
+with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as srv:
+    srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    srv.bind((host, port))
+    srv.listen()
+    while True:
+        conn, addr = srv.accept()
+        threading.Thread(target=handle_client, args=(conn, addr), daemon=True).start()