| import gradio as gr | |
| import requests | |
| import json | |
| import os | |
| def dzen(theme, description=""): | |
| headers = { | |
| 'Content-Type': 'application/json', | |
| 'Authorization': f'Bearer {os.getenv("API_KEY")}' | |
| } | |
| payload = { | |
| 'messages': [{'role': 'system', 'content': f'Напиши пожалуйста классную, понятную, подробную, оригинальную, уникальную статью для Яндекс Дзен, на тему \"{theme}\" {description} Пиши ТОЛЬКО пост (БЕЗ пояснений, БЕЗ markdown, БЕЗ другого текста), текст в посте пиши на языке который используется в описании. Если описание пустое, то сгенерируй интересный пост на любую популярную тему тему.'}], | |
| 'max_tokens': 25000, | |
| 'model': os.getenv("MODEL") | |
| } | |
| response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload) | |
| if response.status_code == 200 and response.text: | |
| try: | |
| data = response.json() | |
| except json.decoder.JSONDecodeError as e: | |
| gr.alert(f'Ошибка при декодировании JSON: {str(e)}') | |
| return '' | |
| 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'] | |
| gr.alert(f'Ошибка: {error_message}') | |
| return '' | |
| else: | |
| gr.alert(f'Не удалось сгенерировать пост. {data}') | |
| return '' | |
| else: | |
| gr.alert(f'Ошибка при получении данных от сервера. Статус код: {response.status_code}') | |
| return '' | |
| iface = gr.Interface(fn=dzen, inputs=[ | |
| gr.Textbox(label="Тема", placeholder=""), | |
| gr.Textbox(label="Дополнительный текст") | |
| ], outputs=gr.Textbox(label="Пост"), title="Генератор постов Яндекс Дзен") | |
| iface.launch() | |