Update ServFire.py

ServFire.py

@@ -8,16 +8,9 @@ import streamlit as st
 import fireworks.client
 
 class WebSocketServer:
-    def __init__(self, host, port):
-        self.host = host
-        self.port = port
+    def __init__(self):
         self.server = None
 
-        db = sqlite3.connect('chat-hub.db')
-        cursor = db.cursor()
-        cursor.execute('CREATE TABLE IF NOT EXISTS messages (id INTEGER PRIMARY KEY AUTOINCREMENT, sender TEXT, message TEXT, timestamp TEXT)')    
-        db.commit()
-    
     async def chatCompletion(self, question):
         
         if "api_key" not in st.session_state:
@@ -113,11 +106,11 @@ class WebSocketServer:
             except Exception as e:
                 print(f"Error: {e}")
 
-    async def start_server(self):
+    async def start_server(self, serverPort):
         self.server = await websockets.serve(
             self.handler,
             self.host,
-            self.port
+            serverPort
         )
         print(f"WebSocket server started at ws://{self.host}:{self.port}")