Update app.py

app.py

@@ -14,10 +14,55 @@ models_list = ["AbsoluteReality 1.8.1", "DALL-E 3 XL", "Playground 2", "Openjour
 
 # PLEASE ❤ like ❤ this space. Please like me. I am 12 years old, one of my projects is: https://ai-hub.rf.gd . I live in Russia, I don't know English very well. Therefore, I apologize that there is only Russian here, but I think it will not be difficult to translate all this. (For example, using gpt)
 
-def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
+def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, gpt):
     if prompt == "" or prompt == None:
         return None
 
+    if gpt:
+        payload = {
+            "model": "gpt-4-vision-preview",
+            "messages": [
+                {
+                    "role": "user",
+                    "content": start,
+                },
+                {
+                    "role": "user",
+                    "content": instruction,
+                }
+            ],
+            "max_tokens": 4095,
+        }    
+
+
+        # API ключ для OpenAI
+        api_key = os.getenv("API_KEY_OPENAI")
+
+        # Заголовки для запроса
+        headers = {
+            'Authorization': f'Bearer {api_key}',
+            'Content-Type': 'application/json',
+        }
+
+        # URL для запроса к API OpenAI
+        url = "https://api.openai.com/v1/chat/completions"
+
+        # Отправляем запрос в OpenAI
+        response = requests.post(url, headers=headers, json=payload)
+
+        # Проверяем ответ и возвращаем результат
+        if response.status_code == 200:
+            response_json = response.json()
+            try:
+                # Пытаемся извлечь текст из ответа
+                return 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
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
@@ -124,6 +169,8 @@ with gr.Blocks(css=css) as dalle:
             strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
         with gr.Row():
             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+        with gr.Row():
+            gpt = gr.Checkbox(label="ChatGPT")
 
     with gr.Tab("Информация"):
         with gr.Row():
@@ -134,6 +181,6 @@ with gr.Blocks(css=css) as dalle:
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, strength], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt, steps, cfg, method, seed, strength, gpt], outputs=image_output)
 
 dalle.launch(show_api=False, share=False)