Update app.py

app.py

@@ -1,13 +1,12 @@
 import os
+import json
 import httpx
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 
-# एप्लीकेशन की नींव
 app = FastAPI()
 
-# CORS पॉलिसी
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -15,54 +14,94 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# सीक्रेट्स से डेटा प्राप्त करना
-API_KEY = os.getenv("SECRET_API_KEY")
-API_URL = os.getenv("SECRET_API_URL")
-MODEL_NAME = os.getenv("SECRET_MODEL_NAME")
-SYSTEM_PROMPT = os.getenv("SECRET_SYSTEM_PROMPT")
+# टोकन और सीक्रेट्स 
 SECRET_TOKEN = "uISssvLLmXK0bsrcFICh14o7aj4q2ZT4"
+API_KEY = os.getenv("SECRET_API_KEY")
+API_URL = os.getenv("SECRET_API_URL", "https://integrate.api.nvidia.com/v1/chat/completions")
+MODEL_NAME = os.getenv("SECRET_MODEL_NAME", "meta/llama-4-maverick-17b-128e-instruct")
+SYSTEM_PROMPT = os.getenv("SECRET_SYSTEM_PROMPT", "You are Aura Gen 2.0, a state-of-the-art AI assistant...")
 
-# 1. रूट हेल्थ चेक (ताकि वह प्लेटफ़ॉर्म सर्वर को ढूंढ सके)
 @app.get("/")
-async def root():
+async def root_health_check():
     return {"status": "Aura Gen 2.0 is online"}
 
-# 2. ओपन एआई फॉर्मेट एंडपॉइंट
+# आप URL में जो भी डालें, यह उसे पकड़ लेगा
 @app.post("/v1/chat/completions")
-async def chat_completions(request: Request):
-    # टोकन वेरिफिकेशन
+@app.post("/")
+async def poe_bot_endpoint(request: Request):
+    # 1. सुरक्षा: आपके टोकन की जांच
     auth_header = request.headers.get("Authorization")
     if auth_header != f"Bearer {SECRET_TOKEN}":
         raise HTTPException(status_code=401, detail="Unauthorized: Invalid Token")
 
     data = await request.json()
-    user_messages = data.get("messages", [])
-    
-    # सिस्टम प्रॉम्ट जोड़ना
-    full_messages = [{"role": "system", "content": SYSTEM_PROMPT}] + user_messages
-    
-    payload = {
-        "model": MODEL_NAME,
-        "messages": full_messages,
-        "max_tokens": data.get("max_tokens", 4096),
-        "temperature": data.get("temperature", 1.0),
-        "stream": True,
-        "chat_template_kwargs": {"enable_thinking": True}
-    }
+    req_type = data.get("type")
+
+    # 2. POE हेल्थ चेक (Settings) - यही 'Run Check' को पास करेगा!
+    if req_type == "settings":
+        return JSONResponse(content={
+            "server_bot_dependencies": {},
+            "allow_attachments": True,
+            "introduction_message": "नमस्ते! मैं Aura Gen 2.0 हूँ। मैं आपकी कैसे सहायता कर सकता हूँ?"
+        })
+
+    # 3. यूज़र का मैसेज (Query) - जब कोई Poe पर आपसे बात करेगा
+    if req_type == "query":
+        poe_messages = data.get("query", [])
+        
+        # Poe के फॉर्मेट को NVIDIA (OpenAI) फॉर्मेट में बदलना
+        openai_messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+        for msg in poe_messages:
+            # Poe में AI को 'bot' कहते हैं, NVIDIA में 'assistant'
+            role = "assistant" if msg.get("role") == "bot" else msg.get("role", "user")
+            openai_messages.append({"role": role, "content": msg.get("content", "")})
+        
+        payload = {
+            "model": MODEL_NAME,
+            "messages": openai_messages,
+            "max_tokens": 4096,
+            "temperature": 1.0,
+            "stream": True,
+            "chat_template_kwargs": {"enable_thinking": True}
+        }
+
+        async def stream_generator():
+            async with httpx.AsyncClient(timeout=120.0) as client:
+                async with client.stream(
+                    "POST", 
+                    API_URL, 
+                    headers={"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"},
+                    json=payload
+                ) as response:
+                    # यदि NVIDIA से कोई एरर आता है
+                    if response.status_code != 200:
+                        yield f"event: text\ndata: {json.dumps({'text': f'⚠️ NVIDIA Server Error: {response.status_code}'})}\n\n"
+                        yield "event: done\ndata: {}\n\n"
+                        return
+
+                    # स्ट्रीमिंग को Poe के फॉर्मेट में बदलना
+                    async for line in response.aiter_lines():
+                        if line.startswith("data: "):
+                            data_str = line[6:].strip()
+                            if data_str == "[DONE]":
+                                continue
+                            try:
+                                parsed = json.loads(data_str)
+                                token = parsed.choices[0].delta.get("content", "")
+                                if token:
+                                    # Poe को यह खास फॉर्मेट चाहिए होता है
+                                    poe_event = {"text": token}
+                                    yield f"event: text\ndata: {json.dumps(poe_event)}\n\n"
+                            except Exception:
+                                pass
+            
+            # स्ट्रीम खत्म होने का संकेत
+            yield "event: done\ndata: {}\n\n"
 
-    async def stream_generator():
-        async with httpx.AsyncClient(timeout=120.0) as client:
-            async with client.stream(
-                "POST", 
-                API_URL, 
-                headers={"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"},
-                json=payload
-            ) as response:
-                async for line in response.aiter_lines():
-                    if line:
-                        yield f"{line}\n\n"
+        # Poe के लिए सही मीडिया टाइप
+        return StreamingResponse(stream_generator(), media_type="text/event-stream")
 
-    return StreamingResponse(stream_generator(), media_type="text/event-stream")
+    return JSONResponse(status_code=400, content={"error": "Unsupported request type"})
 
 if __name__ == "__main__":
     import uvicorn