import os import random import requests import gradio as gr import json css_url = "https://neurixyufi-aihub.static.hf.space/style.css" try: response = requests.get(css_url) response.raise_for_status() css = response.text + " h1{text-align:center}" except requests.exceptions.RequestException as e: print(f"Ошибка загрузки CSS: {e}") css = " h1{text-align:center}" API_URL = "https://text.pollinations.ai/openai" MODEL_NAME = "gemini" # Или какая у вас модель TEMPERATURE = 0.7 MAX_TOKENS = 1200 def call_pollinations_api(messages, temperature=TEMPERATURE, max_tokens=MAX_TOKENS, stream=False): # stream=False """Calls the Pollinations API with the given messages.""" headers = { "Content-Type": "application/json" } data = { "model": MODEL_NAME, "messages": messages, "temperature": temperature, "max_tokens": max_tokens, "stream": stream } response = requests.post(API_URL, headers=headers, data=json.dumps(data)) response.raise_for_status() # Raise HTTPError for bad responses (4xx or 5xx) return response def generate_story(prompt, style): try: messages = [ {"role": "system", "content": f"Напиши хорошую историю в стиле '{style}'. Подробную, понятную, человечную (с душой), уникальную. Не обязательно делать концовку, можно только начало длинной истории. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь, по скольку это ТВОРЧЕСТВО! Не в коем случае не пиши подобное: 'Хорошо, вот ваш текст:', всегда пиши текст, без лишних комментариев."}, {"role": "user", "content": prompt} ] response = call_pollinations_api(messages) full_response = response.json() # Получаем JSON-ответ целиком print("Полный ответ от API:", json.dumps(full_response, indent=2, ensure_ascii=False)) # Выводим отформатированный JSON story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой." yield story except requests.exceptions.RequestException as e: yield f"Ошибка API запроса: {e}" except Exception as e: yield f"Ошибка генерации: {e}" def edit_story(original_story="", edited_prompt=""): try: messages = [ {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания. Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."}, {"role": "user", "content": edited_prompt}, {"role": "user", "content": f"История: ```\n{original_story}\n```"} ] response = call_pollinations_api(messages, max_tokens=32760, stream=False) full_response = response.json() print("Полный ответ от API (edit):", json.dumps(full_response, indent=2, ensure_ascii=False)) edited_story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой." yield edited_story except requests.exceptions.RequestException as e: yield f"Ошибка API запроса: {e}" except Exception as e: yield f"Ошибка редактирования: {e}" def next_story_func(original_story="", next_prompt="", continuation_type="Продолжение"): try: if continuation_type == "Продолжение следует...": continuation_prompt = "Продолжи, подведя к тому, что продолжение следует." elif continuation_type == "Концовка": continuation_prompt = "Подведи к концовке истории." else: continuation_prompt = next_prompt messages = [ {"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения и чтобы было правильно составлено дословно с конца до начала продолжения. (В начале добавляй новые строки/пробел если надо для отступа). Разрешены любые слова, любые темы и стили, что угодно. (В том числе маты и NSFW) Но только если попросит пользователь."}, {"role": "user", "content": continuation_prompt}, {"role": "user", "content": f"История: ```\n{original_story}\n```"} ] response = call_pollinations_api(messages, stream=False) full_response = response.json() print("Полный ответ от API (next):", json.dumps(full_response, indent=2, ensure_ascii=False)) next_story = full_response["choices"][0]["message"]["content"] if "choices" in full_response and len(full_response["choices"]) > 0 else "Ответ не получен или структура ответа не соответствует ожидаемой." yield next_story except requests.exceptions.RequestException as e: yield f"Ошибка API запроса: {e}" except Exception as e: yield f"Ошибка продления: {e}" def edone_story(original_story="", edited_story=""): if edited_story == "": return original_story return edited_story def ndone_story(original_story, next_story_output): return original_story + " " + next_story_output with gr.Blocks(css=css) as demo: gr.Markdown("# Песочница историй") with gr.Row(): with gr.Column(): with gr.Row(): style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Страшная", "Случайный", "Пользовательский"] style = gr.Dropdown(choices=style_choices, label="Выберите стиль истории", value="Приключенческая") with gr.Row(): prompt = gr.Textbox(label="Введите запрос для истории", placeholder="Например: История о путешествии в космос", lines=5) with gr.Row(): generate_button = gr.Button("Создать историю", variant='primary') with gr.Row(): output_story = gr.Textbox(label="История", lines=10, placeholder="Здесь будет ваша новая история!") with gr.Column(): with gr.Accordion("Действия", open=True): with gr.Tab("Редактирование"): edited_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5) edit_button = gr.Button("Отредактировать", variant='primary') edited_story = gr.Textbox(label="Отредактированная история", lines=10, placeholder="Здесь будет ваша новая история!") edone_button = gr.Button("Принять") with gr.Tab("Продление"): continuation_type = gr.Radio(choices=["Продолжение", "Продолжение следует...", "Концовка"], label="Выберите тип продолжения", value="Продолжение") next_prompt = gr.Textbox(label="Введите изменения для продления истории (Необязательно)", placeholder="Продолжи, но что бы было...", lines=5) next_button = gr.Button("Продлить", variant='primary') next_story_output = gr.Textbox(label="Продолжение истории", lines=10, placeholder="Здесь будет продолжение вашей истории!") ndone_button = gr.Button("Принять") generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], concurrency_limit=250) edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], concurrency_limit=250) next_button.click(next_story_func, inputs=[output_story, next_prompt, continuation_type], outputs=[next_story_output], concurrency_limit=250) edone_button.click(edone_story, inputs=[output_story, edited_story], outputs=[output_story], concurrency_limit=550) ndone_button.click(ndone_story, inputs=[output_story, next_story_output], outputs=[output_story], concurrency_limit=550) demo.queue(api_open=False).launch()