adding files

main.py

@@ -169,16 +169,17 @@ logging.info(f"THREADS: {os.getenv('THREADS', 'Not Set')}")
 logging.info(f"CONNECTION_LIMIT: {os.getenv('CONNECTION_LIMIT', 'Not Set')}")
 logging.info(f"CHANNEL_TIMEOUT: {os.getenv('CHANNEL_TIMEOUT', 'Not Set')}")
 
-# Use Flask development server for Hugging Face Spaces compatibility (instead of Waitress)
+# Use Waitress for production server, or Flask for development
 if __name__ == "__main__":
     try:
-        # Log the server startup info
         logging.info("Starting Flask application...")
-
-        # Start the Flask application with the development server for testing
-        app1.run(host="0.0.0.0", port=int(os.getenv("PORT", 7860)), threaded=True)
+        port = int(os.getenv("PORT", 7860))  # Default to 7860 if PORT is not set
+        logging.info(f"Flask app is starting on port {port}...")
+        serve(app1, host="0.0.0.0", port=port, threads=int(os.getenv('THREADS', 6)),
+              connection_limit=int(os.getenv('CONNECTION_LIMIT', 500)), 
+              channel_timeout=int(os.getenv('CHANNEL_TIMEOUT', 120)))
     except Exception as e:
-        logging.error(f"Error starting Flask application: {e}")
+        logging.error(f"Error starting Flask application with Waitress: {e}")
         raise
 
 
@@ -189,3 +190,4 @@ if __name__ == "__main__":
 
 
 
+