app.py

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+from transformers import pipeline
+
+# Ładowanie modelu raz
+chatbot = pipeline("text-generation", model="radlab/polish-gpt2-small", max_new_tokens=80)
+
+# Historia rozmowy
+conversation_history = ""
+
+def respond(user_input, history):
+    global conversation_history
+    conversation_history += f"Użytkownik: {user_input}\nAI: "
+
+    response = chatbot(conversation_history, do_sample=True, top_k=50, top_p=0.95, temperature=0.7)[0]["generated_text"]
+    ai_response = response[len(conversation_history):].split("Użytkownik:")[0].strip()
+
+    conversation_history += ai_response + "\n"
+    return conversation_history, conversation_history
+
+def clear_history():
+    global conversation_history
+    conversation_history = ""
+    return "", ""
+
+# UI
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 Polski Chatbot AI\nNapisz coś po polsku, a AI odpowie!")
+
+    chatbox = gr.Textbox(label="Twoja wiadomość")
+    history_box = gr.Textbox(label="📖 Historia rozmowy", lines=10)
+
+    submit_btn = gr.Button("Wyślij")
+    clear_btn = gr.Button("🧹 Wyczyść")
+
+    submit_btn.click(fn=respond, inputs=[chatbox, history_box], outputs=[history_box, history_box])
+    clear_btn.click(fn=clear_history, outputs=[history_box, history_box])
+
+demo.launch()