fix : serverbug

server/app.py

@@ -29,7 +29,6 @@ def start_generation():
 
 @app.route('/setup', methods=['POST'])
 def setup():
-    global generator
     data = request.get_json()
     model = data.get('model', None)
     if model is None:
@@ -39,7 +38,6 @@ def setup():
 
 @app.route('/get_updated_sentence')
 def get_updated_sentence():
-    global generator
     if not generator:
         return jsonify({"error": "Generator not initialized"}), 400
     return generator.get_status()

server/sources/generator.py

@@ -1,5 +1,4 @@
 
-from flask import jsonify
 import threading
 import logging
 from abc import abstractmethod
@@ -47,7 +46,7 @@ class GeneratorLLM():
     
     def get_status(self) -> dict:
         with self.state.lock:
-            return jsonify(self.state.status())
+            return self.state.status()
 
     @abstractmethod
     def generate(self, history: list) -> None:
@@ -58,4 +57,8 @@ class GeneratorLLM():
         returns:
             None
         """
-        pass
+        pass
+
+if __name__ == "__main__":
+    generator = GeneratorLLM()
+    generator.get_status()