Update app/main.py

app/main.py

@@ -1,6 +1,6 @@
-from fastapi import FastAPI, UploadFile, File, WebSocket, WebSocketDisconnect, HTTPException, Depends
+from fastapi import FastAPI, UploadFile, File, WebSocket, WebSocketDisconnect, HTTPException, Body
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse, JSONResponse
+from fastapi.responses import FileResponse
 from pydantic import BaseModel
 from textblob import TextBlob
 from passlib.context import CryptContext
@@ -8,100 +8,76 @@ from typing import List, Optional
 import os
 import json
 import base64
-from google.oauth2.credentials import Credentials
-from googleapiclient.discovery import build
-from googleapiclient.http import MediaIoBaseUpload
 
 app = FastAPI(title="Zenith Platform")
 
-# --- SECURITY & CONFIG ---
+# --- CONFIG ---
 pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
-GDRIVE_TOKEN_B64 = os.getenv("GDRIVE_TOKEN_B64")
-# Секретный код админа (4 буквы). Установи в Secrets на HF!
-ADMIN_SECRET_CODE = os.getenv("ADMIN_SECRET_CODE", "XAE1") 
+# ВАЖНО: Если переменная не задана, код будет '1234'. Проверь это!
+ADMIN_SECRET_CODE = os.getenv("ADMIN_SECRET_CODE", "1234") 
 
 # --- MODELS ---
-class LoginModel(BaseModel):
-    username: str
-    password: str
-
 class CreateUserModel(BaseModel):
     admin_code: str
     new_username: str
     new_password: str
 
+class UpdateProfileModel(BaseModel):
+    username: str
+    is_looking: bool
+
 class MessageInput(BaseModel):
     text: str
 
-class MoodResponse(BaseModel):
-    mood_color: str
-    emoji: str
-    sentiment: float
-
-# --- GOOGLE DRIVE (Тот же код) ---
-def get_drive_service():
-    try:
-        if not GDRIVE_TOKEN_B64: return None
-        creds_json = base64.b64decode(GDRIVE_TOKEN_B64).decode('utf-8')
-        creds_dict = json.loads(creds_json)
-        creds = Credentials.from_authorized_user_info(creds_dict)
-        return build('drive', 'v3', credentials=creds)
-    except Exception as e:
-        print(f"Drive Error: {e}")
-        return None
-
-# --- AUTH ROUTES ---
-@app.post("/api/auth/login")
-async def login(data: LoginModel):
-    # В реальном проде здесь нужен запрос к Supabase для проверки хэша
-    # Но так как мы делаем "быстрый" бэк, мы будем хэшировать на фронте? 
-    # Нет, правильно: фронт шлет пароль -> бэк проверяет.
-    # Для упрощения работы с Supabase из Python без ORM, 
-    # мы будем делать простую проверку: хэширование пароля для сверки
-    return {"status": "ok", "msg": "Логика проверки перенесена на клиент через Supabase Auth (упрощение)"}
-
+# --- AUTH & ADMIN ---
 @app.post("/api/admin/create_user")
 async def create_user(data: CreateUserModel):
-    if data.admin_code != ADMIN_SECRET_CODE:
-        raise HTTPException(status_code=403, detail="Неверный код доступа")
+    # Логируем для отладки (будет видно в логах HF)
+    print(f"Admin attempt. Input code: '{data.admin_code}', Expected: '{ADMIN_SECRET_CODE}'")
     
-    # Хэшируем пароль
-    hashed_pw = pwd_context.hash(data.new_password)
+    if data.admin_code.strip() != ADMIN_SECRET_CODE:
+        raise HTTPException(status_code=403, detail="ACCESS DENIED: Wrong Code")
     
+    hashed_pw = pwd_context.hash(data.new_password)
     return {
         "username": data.new_username,
         "password_hash": hashed_pw,
         "badge": "BETA"
     }
 
+@app.post("/api/user/update_profile")
+async def update_profile(data: UpdateProfileModel):
+    # Здесь просто возвращаем статус, реальная запись идет через Supabase Client на фронте
+    # или можно расширить этот метод для серверной записи.
+    return {"status": "ok", "looking_for_friends": data.is_looking}
+
 # --- CHAT & MOOD ---
-@app.post("/api/analyze_mood", response_model=MoodResponse)
+@app.post("/api/analyze_mood")
 async def analyze_mood(msg: MessageInput):
     blob = TextBlob(msg.text)
     p = blob.sentiment.polarity
-    if p > 0.5: return MoodResponse(mood_color="#00f2ea", emoji="✨", sentiment=p)
-    elif p > 0: return MoodResponse(mood_color="#4facfe", emoji="😌", sentiment=p)
-    elif p < -0.5: return MoodResponse(mood_color="#ff0055", emoji="🔥", sentiment=p)
-    elif p < 0: return MoodResponse(mood_color="#8e44ad", emoji="🥀", sentiment=p)
-    return MoodResponse(mood_color="#ffffff", emoji="🌫️", sentiment=p)
+    if p > 0.5: return {"mood_color": "#00f2ea", "emoji": "✨"}
+    elif p > 0: return {"mood_color": "#4facfe", "emoji": "😌"}
+    elif p < -0.5: return {"mood_color": "#ff0055", "emoji": "🔥"}
+    elif p < 0: return {"mood_color": "#8e44ad", "emoji": "🥀"}
+    return {"mood_color": "#ffffff", "emoji": "🌫️"}
 
 @app.get("/api/config")
 async def get_config():
-    return {
-        "supabase_url": SUPABASE_URL, 
-        "supabase_key": SUPABASE_KEY
-    }
+    return {"supabase_url": SUPABASE_URL, "supabase_key": SUPABASE_KEY}
 
-# --- WEBSOCKET SIGNALING ---
+# --- WEBSOCKET (FIXED) ---
 class ConnectionManager:
     def __init__(self): self.active_connections: List[WebSocket] = []
     async def connect(self, websocket: WebSocket): await websocket.accept(); self.active_connections.append(websocket)
     def disconnect(self, websocket: WebSocket): self.active_connections.remove(websocket)
     async def broadcast(self, message: str, sender: WebSocket):
         for connection in self.active_connections:
-            if connection != sender: await connection.send_text(message)
+            if connection != sender: 
+                try: await connection.send_text(message)
+                except: pass
 
 manager = ConnectionManager()
 
@@ -115,7 +91,6 @@ async def websocket_endpoint(websocket: WebSocket):
     except WebSocketDisconnect:
         manager.disconnect(websocket)
 
-# --- STATIC ---
 app.mount("/static", StaticFiles(directory="static"), name="static")
 @app.get("/")
 async def read_index(): return FileResponse('static/index.html')