flask_server.py

from flask import Flask, request, jsonify
from flask_cors import CORS
from assistant_agi import process, agent

app = Flask(__name__)
CORS(app)  # Enable CORS for all routes

@app.route('/process', methods=['POST'])
def process_query():
    data = request.json
    if not data or 'query' not in data:
        return jsonify({'error': 'Missing query parameter'}), 400
    
    try:
        response = process(data['query'])
        return jsonify({'response': response})
    except Exception as e:
        return jsonify({'error': str(e)}), 500

@app.route('/chat', methods=['POST'])
def chat_query():
    data = request.json
    if not data or 'query' not in data:
        return jsonify({'error': 'Missing query parameter'}), 400
    
    try:
        response = agent.chat(data['query'])
        return jsonify({'response': response})
    except Exception as e:
        return jsonify({'error': str(e)}), 500

@app.route('/calendar/events', methods=['GET'])
def get_events():
    try:
        # In a real implementation, you would fetch events from your CalDAV server
        # For now, we'll return a simple response
        events = [
            {
                "id": "1",
                "title": "Team Meeting",
                "start": "2025-06-10T10:00:00",
                "end": "2025-06-10T11:00:00",
                "description": "Weekly team sync",
                "location": "Conference Room A"
            },
            {
                "id": "2",
                "title": "Lunch with Client",
                "start": "2025-06-11T12:30:00",
                "end": "2025-06-11T13:30:00",
                "description": "Discuss project requirements",
                "location": "Downtown Cafe"
            },
            {
                "id": "3",
                "title": "Project Deadline",
                "start": "2025-06-13T17:00:00",
                "end": "2025-06-13T17:00:00",
                "description": "Submit final deliverables",
                "location": ""
            }
        ]
        return jsonify({'events': events})
    except Exception as e:
        return jsonify({'error': str(e)}), 500

@app.route('/calendar/events', methods=['POST'])
def create_event():
    data = request.json
    if not data or 'title' not in data or 'start' not in data or 'end' not in data:
        return jsonify({'error': 'Missing required event parameters'}), 400
    
    try:
        # In a real implementation, you would create an event in your CalDAV server
        # For now, we'll just return success
        return jsonify({'success': True, 'id': '4'})
    except Exception as e:
        return jsonify({'error': str(e)}), 500

def run_server():
    """Entry point for the console script"""
    app.run(host='0.0.0.0', port=5000)

if __name__ == '__main__':
    run_server()