Update app.py

app.py

@@ -1,7 +1,33 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Initialize text-generation pipeline with GPT-2 or any HF-compatible model
+generator = pipeline('text-generation', model='gpt2')
+
+def riley_chat(user_input, chat_history=[]):
+    # Riley persona prompt
+    persona = (
+        "You are Riley, an intelligent and warm AI assistant. "
+        "You never mention that you are an AI model. "
+        "Always speak with a futuristic and friendly tone.\n\n"
+        f"User: {user_input}\nRiley:"
+    )
+    output = generator(persona, max_length=100, num_return_sequences=1)
+    riley_response = output[0]['generated_text'][len(persona):].strip()
+
+    chat_history.append((user_input, riley_response))
+    return chat_history, chat_history
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 Riley AI Chat Interface")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Talk to Riley here...")
+    clear = gr.Button("Clear")
+
+    def user_submit(text, history):
+        return "", riley_chat(text, history)
+
+    msg.submit(user_submit, inputs=[msg, chatbot], outputs=[msg, chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 demo.launch()