import csv import os import json import secrets from typing import List, Dict, Union, Annotated from fastapi import FastAPI, Request, HTTPException, Header, Depends from fastapi.responses import HTMLResponse, JSONResponse, FileResponse from fastapi.staticfiles import StaticFiles from pydantic import BaseModel from openai import OpenAI from dotenv import load_dotenv from starlette.middleware.sessions import SessionMiddleware from impresora.printer import PrinterUSB from impresora.order import * # Load environment variables from .env file load_dotenv() #pruebitas4 # Configuration OPENAI_API_KEY = os.getenv("OPENAI_API_KEY") PORT = int(os.getenv("PORT", 6001)) # SECRET_KEY is crucial for signing session cookies. # Fallback to a default if not set, but warn that this is insecure for production. SECRET_KEY = os.getenv("SECRET_KEY", "your_very_very_secret_key_for_signing_cookies_python_v2") if SECRET_KEY == "your_very_very_secret_key_for_signing_cookies_python_v2": print("WARNING: Using default SECRET_KEY. Please set a strong SECRET_KEY in your .env file for production.") if not OPENAI_API_KEY: print("CRITICAL ERROR: OPENAI_API_KEY environment variable not set. The applicaton will not work correctly.") # Potentially exit or prevent app startup if critical env var is missing # raise ValueError("OPENAI_API_KEY is not set, cannot start application.") # --- FastAPI App Initialization --- app = FastAPI(title="Web Pedidos Klein - FastAPI Backend") # Add SessionMiddleware # This middleware adds session support using signed cookies. # Original Express maxAge was 1 hour (60 * 60 * 1000 ms) app.add_middleware( SessionMiddleware, secret_key=SECRET_KEY, max_age=60 * 60 # max_age in seconds for Starlette ) # --- Data Loading --- # Assumes data.json is in the same directory as main.py # The original path was web_pedidos/src/data.json # For the Python version, copy src/data.json to be alongside main.py BG_DATA_PATH = os.path.join(os.path.dirname(__file__), 'data.json') PRODUCTS_PATH = os.path.join(os.path.dirname(__file__), 'products.json') def load_bg_data() -> List[Dict[str, str]]: try: with open(BG_DATA_PATH, 'r', encoding='utf-8') as f: return json.load(f) except FileNotFoundError: print(f"ERROR: Data file not found at {BG_DATA_PATH}. Serving with empty data.") return [] except json.JSONDecodeError: print(f"ERROR: Could not decode JSON from {BG_DATA_PATH}. Serving with empty data.") return [] def load_products() -> List[Dict[str, str]]: try: with open(PRODUCTS_PATH, 'r', encoding='utf-8') as f: return json.load(f) except FileNotFoundError: print(f"ERROR: Data file not found at {PRODUCTS_PATH}. Serving with empty data.") return [] except json.JSONDecodeError: print(f"ERROR: Could not decode JSON from {PRODUCTS_PATH}. Serving with empty data.") return [] bg_data_loaded = load_bg_data() all_products = load_products() # region --- Pydantic Models for Request/Response Typing --- class Message(BaseModel): role: str content: str class ChatCompletionRequest(BaseModel): messages: List[Message] class ItemWeb(BaseModel): id: int name: str quantity: int price: float itemTotal: float class OrderWeb(BaseModel): customerName: str items: List[ItemWeb] totalAmount: float orderDate: str table: int # endregion --- Pydantic Models for Request/Response Typing --- # region --- OpenAI Service Logic --- openai_client = OpenAI(api_key=OPENAI_API_KEY) async def generate_completion(messages_array: List[Message], session_id: str) -> str: if not OPENAI_API_KEY: print("Error: OpenAI API key is not configured.") raise HTTPException(status_code=500, detail="OpenAI API key not configured on server.") print(f"[OpenAI Service Python] Session/Token {session_id} sent: {[msg.model_dump() for msg in messages_array]}") data_for_prompt = [ f'{{"pregunta": "{item.get("q", "")}", "respuesta": "{item.get("ans", "")}"}}' for item in bg_data_loaded ] data_string = "\n".join(data_for_prompt) preprompt = f""" Eres un asistente de el bar klein, tu nombre es camilo klein, usas emojis para responder. y ser carismatico con el cliente. tus responsabilidades son: - Responder preguntas sobre el menu de el bar klein - Proporcionar información sobre el menú de el bar klein - Proporcionar recomendaciones sobre el menú de el bar klein - Proporcionar información sobre la comida de el bar klein - No puedes tomar pedidos de clientes, solo informar - Debes evadir cualquier pregunta que no sea relacionada con el bar klein para esto usaras los siguientes datos: {data_string} """ # processed_messages: List[Dict[str, str]] = [{"role": "system", "content": preprompt}] processed_messages.extend([msg.model_dump() for msg in messages_array]) try: completion = openai_client.chat.completions.create( model="gpt-4o-mini", # messages=processed_messages, # type: ignore (OpenAI lib expects list of specific dicts) temperature=0.3, # ) response_content = completion.choices[0].message.content return response_content if response_content else "-1" # except Exception as e: print(f"Error calling OpenAI: {e}") # Avoid exposing detailed error messages to the client unless necessary raise HTTPException(status_code=500, detail="Error al procesar tu solicitud con OpenAI.") # endregion --- OpenAI Service Logic --- # --- Security/Token Dependency --- async def get_session_token(request: Request) -> Union[str, None]: return request.session.get("antiAbuseToken") async def protect_chat_api( request: Request, x_app_token: Annotated[Union[str, None], Header(alias="X-App-Token")] = None, session_token: Annotated[Union[str, None], Depends(get_session_token)] = None ): # Equivalent to protectChatAPI middleware if not session_token: raise HTTPException(status_code=403, detail="Acceso denegado: Sesión inválida o token no inicializado.") if not x_app_token: raise HTTPException(status_code=401, detail="Acceso denegado: Falta el token X-Chat-Token.") if x_app_token != session_token: # Log this attempt for security monitoring print(f"WARN: Invalid token attempt. Expected: {session_token}, Received: {x_app_token}") raise HTTPException(status_code=403, detail="Acceso denegado: Token inválido.") return True # Protection passed @app.get("/api/get_products", summary="Get products") async def get_products(): return JSONResponse({"products": all_products}) # --- API Endpoints --- @app.get("/api/chat/init-chat", summary="Initialize chat and get anti-abuse token") async def init_chat(request: Request): current_token = request.session.get("antiAbuseToken") if not current_token: new_token = secrets.token_hex(32) request.session["antiAbuseToken"] = new_token # Store in session print(f"Generated new antiAbuseToken for session: {new_token}") return JSONResponse({"chatToken": new_token}) else: # print(f"Using existing antiAbuseToken for session: {current_token}") return JSONResponse({"chatToken": current_token}) class UserCodeRequest(BaseModel): user_code: str @app.post("/api/existsUser", summary="Check if user exists") async def exists_user(request: UserCodeRequest): with open('users.json', 'r') as f: users = json.load(f) for user in users: if user['userCode'] == request.user_code: return JSONResponse({ "success": True, "userName": user['userName'] }) return JSONResponse({ "success": False, "userName": None }) @app.post("/api/printer/order", summary="Printer order", dependencies=[Depends(protect_chat_api)]) async def printer_order(order: OrderWeb): print("Printer order received") print(order) items = order.items table = order.table printer = PrinterUSB(0xfe6,0x811e) print_order = Order(order.customerName,[Item(item.name, item.price, item.quantity) for item in items]) printer.print_order(print_order, table) if not os.path.exists('logs.csv'): with open('logs.csv', 'w', newline='') as f: writer = csv.writer(f) writer.writerow(['userName', 'table', 'orderDate', 'items']) with open('logs.csv', 'a', newline='') as f: writer = csv.writer(f) writer.writerow([order.customerName, order.table, order.orderDate, items]) @app.post("/api/chat/completions", summary="Get chat completions from OpenAI", dependencies=[Depends(protect_chat_api)]) async def chat_completions(request_data: ChatCompletionRequest, request: Request): # Uses session_token (which is the antiAbuseToken) as an identifier for logging session_identifier = request.session.get("antiAbuseToken", "unknown_session") try: openai_response = await generate_completion(request_data.messages, session_identifier) return JSONResponse({"response": openai_response}) except HTTPException as e: # Re-raise HTTPExceptions from called functions raise e except Exception as e: print(f"Unexpected error in /api/chat/completions: {e}") raise HTTPException(status_code=500, detail="Error interno del servidor al procesar el chat.") @app.get("/", response_class=HTMLResponse, include_in_schema=False) async def serve_index_html(): index_path = os.path.join("public", "index.html") if not os.path.exists(index_path): raise HTTPException(status_code=404, detail="public/index.html not found.") return FileResponse(index_path) app.mount("/", StaticFiles(directory="public", html=False), name="public_root_assets") # --- Main Application Runner --- if __name__ == "__main__": if not OPENAI_API_KEY: print("FATAL: OPENAI_API_KEY is not set. OpenAI features will fail.") print("Please create a .env file with OPENAI_API_KEY='your_key_here'") with open(".env", "w") as f: f.write("OPENAI_API_KEY='your_key_here'") print(f"Servidor corriendo en http://localhost:{PORT}") if not os.path.exists(BG_DATA_PATH): print(f"ADVERTENCIA: {BG_DATA_PATH} no encontrado. El asistente de IA no tendrá datos específicos del menú.") else: print(f"Datos del asistente cargados desde: {os.path.abspath(BG_DATA_PATH)}") import uvicorn uvicorn.run(app, host="0.0.0.0", port=PORT)