Update app.py

app.py

@@ -1,149 +1,77 @@
+# app.py
 import gradio as gr
-import random
-import g4f
+from g4f.client import Client
 
-# Список доступных моделей (для примера)
-AVAILABLE_MODELS = [
-    "gpt-3.5-turbo",
-    "gpt-4", 
-    "claude-3-opus",
-    "llama-2-70b",
-    "mixtral-8x7b"
-]
+client = Client()
 
+# --- Обработчик диалога ---
+def respond(message, history):
+    # Добавляем системное сообщение только при старте истории
+    messages = [{"role": "system", "content": ""}]
+    
+    for human, assistant in history:
+        messages.append({"role": "user", "content": human})
+        messages.append({"role": "assistant", "content": assistant})
+    
+    messages.append({"role": "user", "content": message})
 
-def generate_with_model(model_name, prompt):
     try:
-        response = g4f.ChatCompletion.create(
-            model=model_name,
-            messages=[{"role": "user", "content": prompt}],
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=messages
         )
-        return response
+        bot_message = response.choices[0].message.content
     except Exception as e:
-        return f"Ошибка генерации: {str(e)}"
-def generate_response(prompt, selected_model):
-    """Генерация ответа выбранной моделью"""
-    if not prompt:
-        return "Введите запрос для генерации"
-    
-    # Здесь будет реальная интеграция с g4f
-    # response = g4f.ChatCompletion.create(
-    #     model=selected_model,
-    #     messages=[{"role": "user", "content": prompt}]
-    # )
-    
-    # Заглушка для демонстрации
-    response = generate_with_model(selected_model, prompt)
-    return response
+        bot_message = f"Ошибка: {str(e)}"
 
-def clear_all():
-    """Очистка всех полей"""
-    return "", "", "", ""
+    # Возвращаем обновлённую историю + новый ответ
+    return history + [[message, bot_message]]
 
-def random_generate(prompt):
-    """Генерация случайной моделью"""
-    random_model = random.choice(AVAILABLE_MODELS)
-    response = generate_with_model(random_model, prompt)
-    return random_model, response, response, ""
+# --- Интерфейс ---
+with gr.Blocks(title="ESP Brain") as demo:
+    gr.Markdown("## For api")
 
-def vote_for_best(response1, response2, vote):
-    """Оценка лучшего ответа"""
-    if not vote:
-        return "Пожалуйста, выберите какой ответ лучше"
-    return f"Спасибо за оценку! Вы выбрали: {vote}"
+    chatbot = gr.Chatbot(
+        height=600,
+    )
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🧠 Сравнение языковых моделей")
-    gr.Markdown("Тестируйте разные модели и оценивайте качество ответов!")
-    
     with gr.Row():
-        with gr.Column(scale=1):
-            model_selector = gr.Dropdown(
-                choices=AVAILABLE_MODELS,
-                label="Выберите модель для генерации",
-                value=AVAILABLE_MODELS[0]
-            )
-            prompt_input = gr.Textbox(
-                label="Введите ваш запрос",
-                placeholder="Напишите здесь ваш вопрос или промпт...",
-                lines=3
-            )
-            
-            with gr.Row():
-                clear_btn = gr.Button("🧹 Очистить", variant="secondary")
-                skip_btn = gr.Button("⏭️ Пропустить", variant="secondary") 
-                random_btn = gr.Button("🎲 Случайная модель", variant="primary")
-                generate_btn = gr.Button("🚀 Сгенерировать", variant="primary")
+        txt = gr.Textbox(
+            placeholder="Напиши сообщение...",
+            show_label=False,
+            scale=8
+        )
+        submit_btn = gr.Button("Отправить", scale=2)
 
-        with gr.Column(scale=2):
-            response_output = gr.Textbox(
-                label="Сгенерированный ответ",
-                interactive=False,
-                lines=6
-            )
-            
-            with gr.Row():
-                response1_eval = gr.Textbox(
-                    label="Ответ для оценки 1",
-                    interactive=False,
-                    lines=3
-                )
-                response2_eval = gr.Textbox(
-                    label="Ответ для оценки 2", 
-                    interactive=False,
-                    lines=3
-                )
-            
-            vote_radio = gr.Radio(
-                choices=["Ответ 1 лучше", "Ответ 2 лучше", "Оба хороши", "Оба плохи"],
-                label="Какой ответ лучше?",
-                interactive=True
-            )
-            
-            vote_btn = gr.Button("📊 Оценить ответы", variant="primary")
-            vote_result = gr.Textbox(
-                label="Результат оценки",
-                interactive=False
-            )
+    with gr.Row():
+        retry_btn = gr.Button("🔄 Повторить")
+        undo_btn = gr.Button("↩️ Отменить")
+        clear_btn = gr.Button("🗑️ Очистить")
 
-    # Обработчики событий
-    generate_btn.click(
-        fn=generate_response,
-        inputs=[prompt_input, model_selector],
-        outputs=response_output
-    )
-    
-    clear_btn.click(
-        fn=clear_all,
-        outputs=[prompt_input, response_output, response1_eval, response2_eval]
-    )
-    
-    random_btn.click(
-        fn=random_generate,
-        inputs=[prompt_input],
-        outputs=[model_selector, response_output, response1_eval, response2_eval]
-    )
-    
-    skip_btn.click(
-        lambda: [gr.skip(), gr.skip()],
-        outputs=[response1_eval, response2_eval]
-    )
-    
-    vote_btn.click(
-        fn=vote_for_best,
-        inputs=[response1_eval, response2_eval, vote_radio],
-        outputs=vote_result
-    )
+    # Логика
+    txt.submit(fn=respond, inputs=[txt, chatbot], outputs=chatbot)
+    submit_btn.click(fn=respond, inputs=[txt, chatbot], outputs=chatbot)
+
+    def retry_last(history):
+        if history:
+            last_user_msg = history[-1][0]
+            return history[:-1] + [[last_user_msg, None]]  # очищаем ответ
+        return history
+
+    retry_btn.click(fn=retry_last, inputs=chatbot, outputs=chatbot, queue=False)
+
+    def undo_last(history):
+        return history[:-1]
+
+    undo_btn.click(fn=undo_last, inputs=chatbot, outputs=chatbot, queue=False)
 
-    # Инструкция
-    gr.Markdown("""
-    ## Как использовать:
-    1. Выберите модель из списка или используйте случайную
-    2. Введите ваш запрос
-    3. Нажмите "Сгенерировать" для получения ответа
-    4. Сравните два ответа и оцените какой лучше
-    5. Ваши оценки помогут улучшить систему!
-    """)
+    clear_btn.click(lambda: [], outputs=chatbot, queue=False)
 
-# Для запуска на Hugging Face Spaces
-demo.launch(share=True)
+# --- Запуск ---
+if __name__ == "__main__":
+    demo.queue()
+    demo.launch(
+        share=True,
+        ssr_mode=False,
+        debug=True
+    )