Rooni commited on
Commit
6c03f20
·
verified ·
1 Parent(s): 9b87e69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -25
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} .container { max-width: 800px; margin: 0 auto; }"
19
  except requests.exceptions.RequestException as e:
20
  print(f"Ошибка загрузки CSS: {e}")
21
- css = " h1{text-align:center} .container { max-width: 800px; margin: 0 auto; }"
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.Row():
73
- style_choices = ["Приключенческая", "Научно-фантастическая", "Романтическая", "Комедийная", "Трагическая", "Введите свой стиль:"]
74
- style = gr.Dropdown(choices=style_choices, label="Выберите стиль истории", value="Приключенческая")
75
- with gr.Row():
76
- prompt = gr.Textbox(label="Введите запрос для истории", placeholder="Например: История о путешествии в космос", lines=5)
77
- with gr.Row():
78
- generate_button = gr.Button("Сгенерировать историю", variant='primary')
79
- with gr.Row():
80
- output_story = gr.Textbox(label="История", lines=10)
 
 
 
 
 
 
 
 
 
81
 
82
- with gr.Accordion("Действия", open=False):
83
-
84
- with gr.Tab("Редактирование"):
85
- edited_prompt = gr.Textbox(label="Введите изменения для истории", placeholder="Например: Сделай историю более захватывающей", lines=5)
86
- edit_button = gr.Button("Отредактировать", variant='primary')
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)