# app.py (tu archivo de backend Python)
from flask import Flask, request, jsonify
from ollama import Client
from flask_cors import CORS # Para manejar CORS si tu frontend se sirve desde un origen diferente

app = Flask(__name__)
CORS(app) # Habilita CORS para todas las rutas. En producción, especifica origins.

# Configuración de Ollama (igual que la tenías)
ollama_client = Client(
    host='http://192.168.142.245:11434', # Asegúrate de que esta sea la IP correcta
    headers={'x-some-header': 'some-value'} # Si necesitas headers personalizados
)
OLLAMA_MODEL = 'deepseek-r1:1.5b' # El modelo de Ollama que quieres usar

@app.route('/chat', methods=['POST'])
def chat_endpoint():
    try:
        data = request.json
        user_message = data.get('message')

        if not user_message:
            return jsonify({"error": "No message provided"}), 400

        # --- AQUÍ ES DONDE INTEGRARÍAS TU OTRA API DE TERCERO ---
        # Ejemplo:
        # info_from_third_api = get_info_from_third_party_api(user_message)
        #
        # Enriquecer el prompt con la información de la API de tercero
        # enriched_prompt = f"El usuario pregunta: '{user_message}'. Además, esta es información relevante de mi sistema: {info_from_third_api}. Responde al usuario basándote en esta información."
        #
        # Por ahora, usaremos el mensaje original:
        enriched_prompt = user_message

        # Envía el mensaje a Ollama
        response_ollama = ollama_client.chat(
            model=OLLAMA_MODEL,
            messages=[{'role': 'user', 'content': enriched_prompt}],
            stream=False # Para este ejemplo simple, no hacemos streaming al frontend
        )

        ollama_content = response_ollama['message']['content']

        return jsonify({"reply": ollama_content})

    except Exception as e:
        print(f"Error en el backend: {e}")
        return jsonify({"error": str(e)}), 500

if __name__ == '__main__':
    # Ejecuta el backend en un puerto diferente al de Ollama, por ejemplo 5000
    app.run(host='0.0.0.0', port=5000, debug=True)