Create app.py

app.py

@@ -0,0 +1,109 @@
+# =========================================
+# Dialogue System using DialoGPT (Gradio)
+# =========================================
+
+import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+# -----------------------------
+# 1. Load Model & Tokenizer
+# -----------------------------
+MODEL_NAME = "microsoft/DialoGPT-medium"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model = model.to(device)
+
+# -----------------------------
+# 2. Chat Function
+# -----------------------------
+chat_history_ids = None
+
+def chat(user_input, history):
+    global chat_history_ids
+
+    if not user_input.strip():
+        return history, ""
+
+    # Encode user input
+    new_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors="pt").to(device)
+
+    # Append to history
+    if chat_history_ids is not None:
+        bot_input_ids = torch.cat([chat_history_ids, new_input_ids], dim=-1)
+    else:
+        bot_input_ids = new_input_ids
+
+    # Generate response
+    chat_history_ids = model.generate(
+        bot_input_ids,
+        max_length=1000,
+        pad_token_id=tokenizer.eos_token_id,
+        do_sample=True,
+        top_k=50,
+        top_p=0.95,
+        temperature=0.7
+    )
+
+    # Decode response
+    response = tokenizer.decode(
+        chat_history_ids[:, bot_input_ids.shape[-1]:][0],
+        skip_special_tokens=True
+    )
+
+    history.append((user_input, response))
+    return history, ""
+
+# -----------------------------
+# 3. Reset Function
+# -----------------------------
+def reset_chat():
+    global chat_history_ids
+    chat_history_ids = None
+    return [], ""
+
+# -----------------------------
+# 4. Gradio UI
+# -----------------------------
+with gr.Blocks(title="Dialogue System") as app:
+
+    gr.Markdown("## 🤖 AI Dialogue System (Chatbot)")
+    gr.Markdown("Chat with an AI using DialoGPT")
+
+    chatbot = gr.Chatbot()
+
+    with gr.Row():
+        user_input = gr.Textbox(
+            placeholder="Type your message...",
+            show_label=False
+        )
+
+    with gr.Row():
+        send_btn = gr.Button("Send")
+        clear_btn = gr.Button("Clear Chat")
+
+    # Button actions
+    send_btn.click(
+        chat,
+        inputs=[user_input, chatbot],
+        outputs=[chatbot, user_input]
+    )
+
+    user_input.submit(
+        chat,
+        inputs=[user_input, chatbot],
+        outputs=[chatbot, user_input]
+    )
+
+    clear_btn.click(
+        reset_chat,
+        outputs=[chatbot, user_input]
+    )
+
+# -----------------------------
+# 5. Launch
+# -----------------------------
+app.launch()