From 4676b817e94990d10be6adfa1229c564a3ba699d Mon Sep 17 00:00:00 2001
From: martin legrand <martin.legrand@epitech.eu>
Date: Sat, 29 Mar 2025 13:25:26 +0100
Subject: [PATCH] fix : serverbug

---
 server/app.py               | 2 --
 server/sources/generator.py | 9 ++++++---
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/server/app.py b/server/app.py
index d9e7860..668bbfc 100644
--- a/server/app.py
+++ b/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()
diff --git a/server/sources/generator.py b/server/sources/generator.py
index 5753ee0..8c0197f 100644
--- a/server/sources/generator.py
+++ b/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
\ No newline at end of file
+        pass
+
+if __name__ == "__main__":
+    generator = GeneratorLLM()
+    generator.get_status()
\ No newline at end of file