Create main.py

main.py

@@ -0,0 +1,106 @@
+from fastapi import FastAPI, HTTPException, Query, Request
+from pydantic import BaseModel
+from typing import Optional, Dict, List, Any
+import asyncio
+import logging
+
+# Hardcoded cookies (from your request)
+COOKIES = {
+    'pplx.visitor-id': '712960de-28ad-41c8-939f-480b703c70ee',
+    'g_state': '{"i_l":0,"i_ll":1769717814738,"i_b":"gBcJjnhJuupOuCDdBabINoIsf6604w/8Gi6OD0DxmLc","i_e":{"enable_itp_optimization":3}}',
+    '__ps_r': '_',
+    '__ps_lu': 'https://www.perplexity.ai/onboarding?redirect=https%3A%2F%2Fwww.perplexity.ai%2F%3Flogin-source%3DfloatingSignup%26login-new%3Dtrue%23locale%3Den-US  ',
+    '__ps_fva': '1769717851792',
+    '_gcl_au': '1.1.840583653.1769717852',
+    '_fbp': 'fb.1.1769717851924.246154841382399244',
+    'sidebar-upgrade-badge-30day': '5',
+    '__cf_bm': '_nGeJEhOfX9PTCcj8gB6XslTISyXhlE94.i9d0WODBM-1770115842-1.0.1.1-AMoVeKAfZpiY1Ev9KP1TYcabA.zmHBKxvS6jbCw7jLmtg0u.3fS8Ra8NhXYbzdGKVjcFOUXeDvXg40vu3OXZxqHfrKSHnTjbjr_23NhYN14',
+    '__cflb': '02DiuDyvFMmK5p9jVbVnMNSKYZhUL9aGmrTbYPN5s7xRA',
+    'pplx.session-id': '9934273c-4f6b-4f7f-b092-75681a0100fd',
+    'cf_clearance': 'MvVM3iTW57M3un0OPJ1lE92Z9VUadUjxdF1HFsTnH3Y-1770115847-1.2.1.1-.ZPX_G0dgDxzAaxifqqmcT.aWc6ck916zGTYOHhvqWZA8KdJbCwveBFkJh57NRIpsHIZ.ZWN3FQRQ4dNq.j6K6c0mcqDtoBs5ihuLwM_dw8kQr7Rdox99PlTpCZTv3cH8hIVfZpmvGToLBlk8t5pyGjTD0MOoG8GTWoM0xZzcTfVA4Pm_6m67NJp7nZmzftoxmyELzXrZrK8uoJpIQ06T7aslgbbA4RAHVgNWx6pwFk',
+    '__ps_sr': '_',
+    '__ps_slu': 'https://www.perplexity.ai/search/hi-Km1fNgiUTIiQBPHPXPPDRw  ',
+    '_dd_s': 'aid=b3cdfc91-8ae5-42ca-8313-fd3464265b7c&rum=2&id=3d5d5ff2-38b9-4dd1-987f-dd4348eb1093&created=1770115845969&expire=1770116989591&logs=0',
+    'pplx.metadata': '{%22qc%22:4%2C%22qcu%22:3%2C%22qcm%22:0%2C%22qcc%22:0%2C%22qcco%22:0%2C%22qccol%22:0%2C%22qcdr%22:0%2C%22qcs%22:0%2C%22qcd%22:0%2C%22hli%22:true%2C%22hcga%22:false%2C%22hcds%22:false%2C%22hso%22:true%2C%22hfo%22:true%2C%22hsco%22:false%2C%22hfco%22:false%2C%22hsma%22:false%2C%22hdc%22:false%2C%22hdttb%22:false%2C%22fqa%22:1769717863030%2C%22lqa%22:1770115870300}',
+    '__Secure-next-auth.session-token': 'eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0..envLMas9-Txsy_I6.6Di02enfpLyp0SqjvIqvOjeBrAh-Ua7OT95mPaC_Bm4hU-CLu9vf6xGEhwgcOFiwPkPIAXEYcZ7iAEaqiHlL7tf6RPHo17bICsuOApRNkP5f9C4o_bRG4ZApfDmNq9iYpoadgkojzEiFEy9XdGG_-fjMMdevzQ_LQNPb-URfJwmXWrUtPB2durHzvRJkjPmrndyHih7ljDKWfQMSszxz9LIkEjR4QyPGntrXHYCyStIC_qg13RFPLR10SZyptDKLiWM.OlKoZQWB0wzsIODRxV28fA',
+    '_rdt_uuid': '1769717851824.81056fd1-32a5-4a91-b6e8-9bc24110b278',
+}
+
+EXPECTED_API_KEY = "11223344"
+
+app = FastAPI(title="Perplexity AI Proxy", version="1.0")
+
+logging.basicConfig(level=logging.INFO)
+
+class SearchRequest(BaseModel):
+    query: str
+    mode: str = "auto"
+    model: Optional[str] = None
+    sources: List[str] = ["web"]
+    language: str = "en-US"
+    incognito: bool = False
+
+@app.post("/search")
+async def search_post(request: Request, body: SearchRequest):
+    api_key = request.headers.get("x-api-key")
+    if api_key != EXPECTED_API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
+    try:
+        # Import AFTER installation (now available as local editable package)
+        import perplexity_async
+        client = await perplexity_async.Client(cookies=COOKIES)
+        resp = await client.search(
+            query=body.query,
+            mode=body.mode,
+            model=body.model,
+            sources=body.sources,
+            files={},
+            stream=False,
+            language=body.language,
+            follow_up=None,
+            incognito=body.incognito
+        )
+        return {"success": True, "response": resp}
+    except Exception as e:
+        logging.error(f"Error: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/search")
+async def search_get(
+    request: Request,
+    q: str = Query(..., description="Question"),
+    api_key: Optional[str] = None,
+    mode: str = "auto",
+    model: Optional[str] = None,
+    sources: str = "web",
+    language: str = "en-US",
+    incognito: bool = False
+):
+    key = request.headers.get("x-api-key") or api_key
+    if key != EXPECTED_API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
+    try:
+        import perplexity_async
+        client = await perplexity_async.Client(cookies=COOKIES)
+        sources_list = [s.strip() for s in sources.split(",")] if sources else ["web"]
+        resp = await client.search(
+            query=q,
+            mode=mode,
+            model=model,
+            sources=sources_list,
+            files={},
+            stream=False,
+            language=language,
+            follow_up=None,
+            incognito=incognito
+        )
+        return {"success": True, "response": resp}
+    except Exception as e:
+        logging.error(f"Error: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/health")
+async def health():
+    return {"status": "ok"}