Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,10 +15,10 @@ css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
|
|
15 |
try:
|
16 |
response = requests.get(css_url)
|
17 |
response.raise_for_status()
|
18 |
-
css = response.text + " h1{text-align:center}
|
19 |
except requests.exceptions.RequestException as e:
|
20 |
print(f"Ошибка загрузки CSS: {e}")
|
21 |
-
css = " h1{text-align:center}
|
22 |
|
23 |
def generate_story(prompt, style):
|
24 |
try:
|
@@ -52,7 +52,7 @@ def next_story_func(original_story, next_prompt):
|
|
52 |
try:
|
53 |
client = InferenceClient(api_key=get_random_api_key())
|
54 |
messages = [
|
55 |
-
{"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения."},
|
56 |
{"role": "user", "content": next_prompt},
|
57 |
{"role": "assistant", "content": original_story}
|
58 |
]
|
@@ -69,29 +69,30 @@ def ndone_story(original_story, next_story_output):
|
|
69 |
return original_story + next_story_output
|
70 |
|
71 |
with gr.Blocks(css=css) as demo:
|
72 |
-
with gr.
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
81 |
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
edited_story = gr.Textbox(label="Отредактированная история", lines=10)
|
88 |
-
edone_button = gr.Button("Принять")
|
89 |
-
|
90 |
-
with gr.Tab("Продление"):
|
91 |
-
next_prompt = gr.Textbox(label="Введите изменения для продления истории (Необязательно)", placeholder="Продолжи, но чтобы было...", lines=5)
|
92 |
-
next_button = gr.Button("Продлить", variant='primary')
|
93 |
-
next_story_output = gr.Textbox(label="Продолжение истории", lines=10)
|
94 |
-
ndone_button = gr.Button("Принять")
|
95 |
|
96 |
generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], concurrency_limit=250)
|
97 |
edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], concurrency_limit=250)
|
|
|
15 |
try:
|
16 |
response = requests.get(css_url)
|
17 |
response.raise_for_status()
|
18 |
+
css = response.text + " h1{text-align:center}"
|
19 |
except requests.exceptions.RequestException as e:
|
20 |
print(f"Ошибка загрузки CSS: {e}")
|
21 |
+
css = " h1{text-align:center}"
|
22 |
|
23 |
def generate_story(prompt, style):
|
24 |
try:
|
|
|
52 |
try:
|
53 |
client = InferenceClient(api_key=get_random_api_key())
|
54 |
messages = [
|
55 |
+
{"role": "system", "content": "Продли историю, учитывая предоставленные указания. Продливай В ТОЧНОСТИ С КОНЦА, прям с того же символа, слова, предложения. (В начале добавляй новые строки если надо для отступа)"},
|
56 |
{"role": "user", "content": next_prompt},
|
57 |
{"role": "assistant", "content": original_story}
|
58 |
]
|
|
|
69 |
return original_story + next_story_output
|
70 |
|
71 |
with gr.Blocks(css=css) as demo:
|
72 |
+
with gr.Column():
|
73 |
+
with gr.Row():
|
74 |
+
style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Введите свой стиль:"]
|
75 |
+
style = gr.Dropdown(choices=style_choices, label="Выберите стиль истории", value="Приключенческая")
|
76 |
+
with gr.Row():
|
77 |
+
prompt = gr.Textbox(label="Введите запрос для истории", placeholder="Например: История о путешествии в космос", lines=5)
|
78 |
+
with gr.Row():
|
79 |
+
generate_button = gr.Button("Сгенерировать историю", variant='primary')
|
80 |
+
with gr.Row():
|
81 |
+
output_story = gr.Textbox(label="История", lines=10)
|
82 |
+
|
83 |
+
with gr.Column():
|
84 |
+
with gr.Accordion("Действия", open=True):
|
85 |
+
with gr.Tab("Редактирование"):
|
86 |
+
edited_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
|
87 |
+
edit_button = gr.Button("Отредактировать", variant='primary')
|
88 |
+
edited_story = gr.Textbox(label="Отредактированная история", lines=10)
|
89 |
+
edone_button = gr.Button("Принять")
|
90 |
|
91 |
+
with gr.Tab("Продление"):
|
92 |
+
next_prompt = gr.Textbox(label="Введите изменения для продления истории (Необязательно)", placeholder="Продолжи, но чтобы было...", lines=5)
|
93 |
+
next_button = gr.Button("Продлить", variant='primary')
|
94 |
+
next_story_output = gr.Textbox(label="Продолжение истории", lines=10)
|
95 |
+
ndone_button = gr.Button("Принять")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
96 |
|
97 |
generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], concurrency_limit=250)
|
98 |
edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], concurrency_limit=250)
|