Update app.py

app.py

@@ -20,13 +20,21 @@ class ChatRequest(BaseModel):
     message: str
     reset_history: bool = False  # Флаг для очистки истории
 
+# Модель данных для инициализации
+class InitializeRequest(BaseModel):
+    model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct"
+    system_message: str = "You are a friendly Chatbot."
+    max_tokens: int = 1024
+    temperature: float = 0.7
+    top_p: float = 0.95
+
 # Инициализация модели
 def initialize_model(
-    model_name: str = "Qwen/Qwen2.5-Coder-32B-Instruct",
-    sys_message: str = "You are a friendly Chatbot.",
-    tokens: int = 1024,
-    temp: float = 0.7,
-    top_p_value: float = 0.95,
+    model_name: str,
+    sys_message: str,
+    tokens: int,
+    temp: float,
+    top_p_value: float,
 ):
     global client, system_message, max_tokens, temperature, top_p
     client = InferenceClient(model_name)
@@ -35,6 +43,21 @@ def initialize_model(
     temperature = temp
     top_p = top_p_value
 
+# Маршрут для инициализации модели
+@app.post("/initialize")
+async def initialize(request: InitializeRequest):
+    try:
+        initialize_model(
+            model_name=request.model_name,
+            sys_message=request.system_message,
+            tokens=request.max_tokens,
+            temp=request.temperature,
+            top_p_value=request.top_p,
+        )
+        return {"status": "Model initialized successfully."}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 # Маршрут для обработки запросов
 @app.post("/chat")
 async def chat(request: ChatRequest):
@@ -91,7 +114,13 @@ async def get_status():
     }
 
 # Инициализация модели при запуске
-initialize_model()
+initialize_model(
+    model_name="Qwen/Qwen2.5-Coder-32B-Instruct",
+    sys_message="You are a friendly Chatbot.",
+    tokens=1024,
+    temp=0.7,
+    top_p_value=0.95,
+)
 
 # Запуск приложения (для локального тестирования)
 if __name__ == "__main__":