Update app.py

app.py

@@ -1,9 +1,28 @@
 import gradio as gr
 import os 
 
-def gen(prompt):
-    return f'WORK: {prompt}'
+API_TOKEN = os.getenv("HF_READ_TOKEN")
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+
+def gen(prompt, model='gpt'):
+    if prompt == "" or prompt == None:
+        return None
+
+    if model == 'gpt':
+        API_URL = "https://api-inference.huggingface.co/models/openchat/openchat_3.5"
     
+    payload = {
+        "messages": [{"role": "user", "content": prompt}]
+        }
+
+    response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
+    if response.status_code != 200:
+        print(f"Ошибка: Не удалось получить ответ. Статус ответа: {response.status_code}")
+        print(f"Содержимое ответа: {response.text}")
+        return "Произошла ошибка"
+    
+    return response.json()
+
 
 css = """
 footer {visibility: hidden !important;}