Update app.py

app.py

@@ -24,11 +24,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             "messages": [
                 {
                     "role": "user",
-                    "content": start,
+                    "content": "В этом чате тебе будут присылать prompt для нейросетей генераторов изображений. А ты должен будешь пожалуйста улучшить этот prompt.",
                 },
                 {
                     "role": "user",
-                    "content": instruction,
+                    "content": prompt,
                 }
             ],
             "max_tokens": 4095,
@@ -55,13 +55,11 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
             response_json = response.json()
             try:
                 # Пытаемся извлечь текст из ответа
-                return response_json["choices"][0]["message"]["content"]
+                prompt = response_json["choices"][0]["message"]["content"]
             except Exception as e:
                 # Если есть ошибка в структуре JSON, выводим ее
-                return f"Error processing the image response: {e}"
         else:
             # Если произошла ошибка, возвращаем сообщение об ошибке
-            return f"Error: {response.status_code} - {response.text}"
         
     
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN"), os.getenv("HF_READ_TOKEN_2"), os.getenv("HF_READ_TOKEN_3"), os.getenv("HF_READ_TOKEN_4"), os.getenv("HF_READ_TOKEN_5")]) # it is free