app.py

"""
Flask Web Interface for ITER Bhubaneswar Chatbot
Simple web interface for testing the intelligent chatbot
"""

from flask import Flask, render_template, request, jsonify
from iter_chatbot import ITERChatbot
import os

app = Flask(__name__)

# Initialize chatbot
chatbot = None

def initialize_chatbot():
    """Initialize the chatbot globally"""
    global chatbot
    json_path = '/home/gaurav/Downloads/iter_bhubaneswar_knowledge.json'
    
    if not os.path.exists(json_path):
        print(f"Error: Knowledge base file not found at {json_path}")
        return False
    
    chatbot = ITERChatbot(json_path)
    return chatbot.initialize()

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

@app.route('/chat', methods=['POST'])
def chat():
    """Handle chat requests"""
    if not chatbot:
        return jsonify({'error': 'Chatbot not initialized'}), 500
    
    data = request.get_json()
    user_message = data.get('message', '').strip()
    
    if not user_message:
        return jsonify({'error': 'Empty message'}), 400
    
    try:
        response = chatbot.chat(user_message)
        return jsonify({'response': response})
    except Exception as e:
        return jsonify({'error': f'Chat error: {str(e)}'}), 500

@app.route('/status')
def status():
    """Check chatbot status"""
    return jsonify({
        'status': 'ready' if chatbot else 'not initialized',
        'message': 'Chatbot is ready!' if chatbot else 'Chatbot needs initialization'
    })

if __name__ == '__main__':
    # Create templates directory
    os.makedirs('templates', exist_ok=True)
    
    print("Initializing ITER Chatbot...")
    if initialize_chatbot():
        print("✅ Chatbot ready!")
        print("🌐 Starting web interface...")
        app.run(debug=True, host='0.0.0.0', port=5000)
    else:
        print("❌ Failed to initialize chatbot")