fix : none generator error

server/app.py

@@ -49,7 +49,9 @@ def setup():
 
 @app.route('/get_updated_sentence')
 def get_updated_sentence():
-    return jsonify(generator.get_status())
+    if not generator:
+        return jsonify({"error": "Generator not initialized"}), 400
+    return generator.get_status()
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', threaded=True, debug=True, port=args.port)

server/sources/generator.py

@@ -41,7 +41,6 @@ class GeneratorLLM():
     
     def get_status(self) -> dict:
         with self.state.lock:
-            print("Returning status:\n", self.state.status())
             return jsonify(self.state.status())
 
     @abstractmethod

sources/llm_provider.py

@@ -121,7 +121,6 @@ class Provider:
             is_complete = False
             while not is_complete:
                 response = requests.get(f"http://{self.server_ip}/get_updated_sentence")
-                print("raw:", response.json())
                 thought = response.json()["sentence"]
                 is_complete = bool(response.json()["is_complete"])
                 time.sleep(2)