fix : pretty_print issue

server/app.py

@@ -20,29 +20,28 @@ generator = OllamaLLM() if args.provider == "ollama" else LlamacppLLM()
 @app.route('/generate', methods=['POST'])
 def start_generation():
     if generator is None:
-        return jsonify({"error": "Generator not initialized"}), 400
+        return jsonify({"error": "Generator not initialized"}), 401
     data = request.get_json()
     history = data.get('messages', [])
     if generator.start(history):
         return jsonify({"message": "Generation started"}), 202
-    return jsonify({"error": "Generation already in progress"}), 400
+    return jsonify({"error": "Generation already in progress"}), 402
 
 @app.route('/setup', methods=['POST'])
 def setup():
     data = request.get_json()
     model = data.get('model', None)
     if model is None:
-        return jsonify({"error": "Model not provided"}), 400
+        return jsonify({"error": "Model not provided"}), 403
     generator.set_model(model)
     return jsonify({"message": "Model set"}), 200
 
 @app.route('/get_complete_sentence', methods=['GET'])
 def get_complete_sentence():
     if not generator:
-        return jsonify({"error": "Generator not initialized"}), 400
+        return jsonify({"error": "Generator not initialized"}), 404
     while True:
         status = generator.get_status()
-        print(status)
         if status["is_complete"]:
             return jsonify(status)
     return None
@@ -50,7 +49,7 @@ def get_complete_sentence():
 @app.route('/get_updated_sentence')
 def get_updated_sentence():
     if not generator:
-        return jsonify({"error": "Generator not initialized"}), 400
+        return jsonify({"error": "Generator not initialized"}), 405
     return generator.get_status()
 
 if __name__ == '__main__':

server/sources/generator.py

@@ -40,6 +40,7 @@ class GeneratorLLM():
         with self.state.lock:
             if self.state.is_generating:
                 return False
+            self.state.is_generating = True
             self.logger.info("Starting generation")
             threading.Thread(target=self.generate, args=(history,)).start()
         return True