import os
import psycopg2
from fastapi import FastAPI, Request

app = FastAPI()

# Берем данные из переменных окружения
DATABASE_URL = os.getenv("DATABASE_URL")

@app.post("/webhook")
async def handle_feedback(request: Request):
    data = await request.json()
    
    # 1. ТЕПЕРЬ МЫ ИЗВЛЕКАЕМ ВСЁ
    business_name = data.get("business_name", "Unknown")
    branch = data.get("branch")         # Добавлено
    contact = data.get("contact")       # Добавлено
    raw_message = data.get("raw_message") # Добавлено
    language = data.get("language")     # Добавлено
    
    risk_level = data.get("risk_level", "Low")
    category = data.get("category", "Other")
    summary = data.get("summary", "")
    sentiment_score = data.get("sentiment_score", 0)
    action_required = data.get("action_required", "")

    try:
        conn = psycopg2.connect(DATABASE_URL)
        cur = conn.cursor()
        
        # 2. ОБНОВЛЕННЫЙ ЗАПРОС (Добавлены 4 колонки)
        query = """
        INSERT INTO feedback (
            business_name, branch, contact, raw_message, language, 
            risk_level, category, summary, sentiment_score, action_required
        )
        VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
        """
        
        # 3. ПЕРЕДАЕМ ВСЕ 10 ПАРАМЕТРОВ
        cur.execute(query, (
            business_name, branch, contact, raw_message, language,
            risk_level, category, summary, sentiment_score, action_required
        ))
        
        conn.commit()
        cur.close()
        conn.close()
        
        return {"status": "success", "message": "Feedback saved to Neon"}
    except Exception as e:
        return {"status": "error", "message": str(e)}

@app.get("/")
def read_root():
    return {"message": "Python Bridge is Running!"}