from flask import Flask, render_template, request
import gradio as gr
from transformers import pipeline

app = Flask(__name__)

# Load chatbot model
chatbot = pipeline("text2text-generation", model="facebook/blenderbot-2B")

@app.route('/')
def home():
    return render_template('index.html')

@app.route('/chat')
def chat():
    return render_template('chat.html')

@app.post('/generate_text')
def generate_text(input_text):
    response = chatbot(input_text)
    return response[0]['generated_text']

if __name__ == '__main__':
    app.run(debug=True)