Update app.py

app.py

@@ -4,33 +4,38 @@ import json
 import os
 
 def generate_minecraft_command(input_data):
-    if isinstance(input_data, str):
-        description = input_data
-    else:
-        description = input_data.read()
+    try:
+        if isinstance(input_data, str):
+            description = input_data
+        else:
+            description = input_data.read()
 
-    headers = {
-        'Content-Type': 'application/json',
-        'Authorization': f'Bearer {os.getenv("API_KEY")}'
-    }
+        headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {os.getenv("API_KEY")}'
+        }
 
-    payload = {
-        'messages': [{'role': 'system', 'content': f'{description}'}],
-        'max_tokens': 10000,
-        'model': os.getenv("MODEL")
-    }
+        payload = {
+            'messages': [{'role': 'system', 'content': f'{description}'}],
+            'max_tokens': 10000,
+            'model': os.getenv("MODEL")
+        }
 
-    response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
-    data = json.loads(response.text)
+        response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
+        response.raise_for_status()  # Raises an HTTPError if the HTTP request returned an unsuccessful status code
 
-    if 'choices' in data and len(data['choices']) > 0:
-        command = data['choices'][0]['message']['content'].strip()
-        return command
-    elif 'error' in data:
-        error_message = data['error']['message']
-        return f'Ошибка: {error_message}'
-    else:
-        return f'Не удалось сгенерировать команду. {data}'
+        data = json.loads(response.text)
+
+        if 'choices' in data and len(data['choices']) > 0:
+            command = data['choices'][0]['message']['content'].strip()
+            return command
+        elif 'error' in data:
+            error_message = data['error']['message']
+            return f'Ошибка: {error_message}'
+        else:
+            return f'Не удалось сгенерировать команду. {data}'
+    except Exception as e:
+        return f'Произошла ошибка: {str(e)}'
 
 iface = gr.Interface(
     fn=generate_minecraft_command,