Upload app.py

app.py

@@ -0,0 +1,117 @@
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+TEXT_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
+IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"
+
+
+# ── ЧАТ ──────────────────────────────────────────────────────────────────────
+def respond(message, history, system_prompt, max_tokens, temperature, hf_token: gr.OAuthToken):
+    client = InferenceClient(token=hf_token.token)
+
+    messages = [{"role": "system", "content": system_prompt}]
+    for h in history:
+        messages.append({"role": h["role"], "content": h["content"]})
+    messages.append({"role": "user", "content": message})
+
+    response = ""
+    for chunk in client.chat_completion(
+        messages,
+        model=TEXT_MODEL,
+        max_tokens=max_tokens,
+        temperature=temperature,
+        stream=True,
+    ):
+        token = chunk.choices[0].delta.content or ""
+        response += token
+        yield response
+
+
+# ── ГЕНЕРАЦИЯ ФОТО ────────────────────────────────────────────────────────────
+def generate_image(prompt, steps, hf_token: gr.OAuthToken):
+    if not prompt.strip():
+        return None
+    client = InferenceClient(token=hf_token.token)
+    image = client.text_to_image(
+        prompt,
+        model=IMAGE_MODEL,
+        num_inference_steps=steps,
+    )
+    return image
+
+
+# ── ИНТЕРФЕЙС ─────────────────────────────────────────────────────────────────
+css = """
+footer { display: none !important; }
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=css, title="AI Ассистент") as demo:
+
+    gr.Markdown("# 🤖 AI Ассистент")
+    gr.Markdown("Войди через HuggingFace — это бесплатно, ключи не нужны.")
+
+    with gr.Sidebar():
+        gr.LoginButton()
+        gr.Markdown("---")
+        gr.Markdown("**Модели:**")
+        gr.Markdown(f"💬 `{TEXT_MODEL}`")
+        gr.Markdown(f"🎨 `{IMAGE_MODEL}`")
+
+    with gr.Tabs():
+
+        # ── Вкладка: Чат ──────────────────────────────────────────────────────
+        with gr.Tab("💬 Чат"):
+            gr.ChatInterface(
+                respond,
+                type="messages",
+                chatbot=gr.Chatbot(height=480, render_markdown=True, show_label=False),
+                textbox=gr.Textbox(placeholder="Напиши что-нибудь...", container=False),
+                additional_inputs=[
+                    gr.Textbox(
+                        value="Ты умный и дружелюбный ассистент. Отвечай на языке пользователя.",
+                        label="Системный промпт",
+                        lines=2,
+                    ),
+                    gr.Slider(256, 2048, value=512, step=128, label="Макс. токенов"),
+                    gr.Slider(0.1, 1.5, value=0.7, step=0.05, label="Температура"),
+                ],
+                additional_inputs_accordion=gr.Accordion("⚙️ Настройки", open=False),
+                submit_btn="Отправить ➤",
+                retry_btn="🔄",
+                undo_btn="↩️",
+                clear_btn="🗑️",
+            )
+
+        # ── Вкладка: Генерация фото ────────────────────────────────────────────
+        with gr.Tab("🎨 Генерация фото"):
+            with gr.Row():
+                with gr.Column(scale=2):
+                    prompt_box = gr.Textbox(
+                        label="Описание картинки",
+                        placeholder="закат на море, аниме стиль, 4K...",
+                        lines=4,
+                    )
+                    steps_slider = gr.Slider(1, 8, value=4, step=1, label="Шаги (быстрее = меньше)")
+                    gen_btn = gr.Button("🎨 Создать", variant="primary", size="lg")
+
+                with gr.Column(scale=3):
+                    output_image = gr.Image(label="Результат", show_label=False)
+
+            gen_btn.click(
+                fn=generate_image,
+                inputs=[prompt_box, steps_slider],
+                outputs=output_image,
+            )
+
+            gr.Examples(
+                examples=[
+                    ["a futuristic cyberpunk city at night, neon lights, rain, 4K"],
+                    ["cute anime girl in a flower field, soft light, detailed"],
+                    ["astronaut riding a horse on Mars, cinematic, realistic"],
+                    ["a cozy cabin in snowy forest, warm light inside, peaceful"],
+                ],
+                inputs=prompt_box,
+            )
+
+if __name__ == "__main__":
+    demo.launch()