o HiV*@s`ddlmZmZmZmZmZddlmZmZm Z ddl m Z ddl Z ddl m Z ddlmZddlmZddlmZdd lmZmZmZmZmZmZmZmZeZe eZ e !e j"eej#eefd efd d Z$ej%d ddeej#eefddZ&ej'dddeej#fddZ(ej'dddeej#ee$fde)defddZ*ej'dddeej#fde)de)fdd Z+ej%d!d"ded#eej#ee$eefde)d$ee)efdefd%d&Z,ej%d'd(ded#eej#ee$eefd$ee)efdefd)d*Z-e%d+eej#eefd$e.fd,d-Z/e%d.d$e.fd/d0Z0e'd1eej#eefd2d3Z1ej%d4d5ded#fd$ee)effd6d0Z0ej%d7d8ded#eej#fd9e)d$ee)effd:d;Z2ej'dd?Z3ej%d@dAdedBdCifd$ee)effdDdEZ4ej'dFdGdedHdIdJedKdLdJeej#eeee$fdMe)dNe5defdOdPZ6dS)Q) APIRouterDepends HTTPExceptionBodyQuery)DictAnyOptional)datetimeN)ObjectId)database)get_current_userdetails) ZerodhaClient)refresh_movers_serviceget_top10_servicecreate_snapshot_serviceget_snapshot_servicechat_symbol_serviceplace_short_order_servicekite_postback_serviceget_live_signals_servicereturncCsNt|d}|dd|i}|stdddt|d|d|d d d S) zv Build Zerodha client for the current user from DB settings. Throws HTTPException(404) if not configured. _idzerodha_settingsuser_idz#Zerodha settings not found for user status_codedetailapi_key api_secret access_token)rr r!)strgetfind_onerr)db current_userrsettingsr);/var/www/html/Trade-python/app/v1/routers/tradeengineGPT.pyget_zerodha_client_from_dbs  r+z/movers/refreshz7Refresh movers aggregator (poll sources, merge, dedupe))summaryc CsVz t|}d|dWStyty*}z tdtdt|dd}~ww)Nokstatusresultzrefresh_movers failedr)rr Exceptionlogger exceptionr#)r&r'r0er)r)r*refresh_movers1   r6z /movers/top10zReturn Top10 movers (cached)c CVz t|}d|dWStyty*}z tdtdt|dd}~ww)Nr-)r/top10zget_top10 failedr1r)rrr2r3r4r#)r&topr5r)r)r*r9=r7r9z/symbol/{symbol}/snapshotz-Create and return fresh snapshot for a symbolsymbolzerodhac Cshzt|||}dt|d|ddWStyty3}z tdtdt|dd}~ww)Nr-rsnapshot)r/ snapshot_idr=zcreate_snapshot failedr1r)rr#rr2r3r4)r;r&r<docr5r)r)r*create_snapshotJs   r@z'/symbol/{symbol}/snapshot/{snapshot_id}zGet existing snapshot by idr>c Cshzt||}|stdddd|dWStyty3}z tdtdt|dd}~ww)NrzSnapshot not foundrr-)r/r=zget_snapshot failedr1)rrr2r3r4r#)r;r>r&r?r5r)r)r* get_snapshotZs     rAz/symbol/{symbol}/chatz0Ask LLM about a symbol using snapshot + question.payloadc Csbz|}t|||||}d|dWStyty0}z tdtdt|dd}~ww)zt payload: { "question": str, "snapshot_id": Optional[str] } Returns: { status: ok, response: LLM_JSON } r-)r/responsezsymbol_chat failedr1rN)rrr2r3r4r#)r;rBr&r<r'userrespr5r)r)r* symbol_chatis    rFz /orders/shortz>Place manual short order (requires explicit user_confirmation)c Cs`z|}t||||}d|dWStyty/}z tdtdt|dd}~ww)z\ payload: { symbol, quantity, price(optional), order_type, user_confirmation:true } r-)r/placedzplace_short_order failedr1rN)rrr2r3r4r#)rBr&r<r'rDr0r5r)r)r*place_short_orders    rHz /callbackcCs|d}t|d}|stddd|dd|i}|s%tdd dt|d |d }||}|dd|id d |d iiddiS)N request_tokenrzMissing request tokenrrrrzZerodha settings not foundrr $setr!r/ connected)r$r#rr%rgenerate_session update_one)rBr&r'rIrr(clientsessionr)r)r* post_callbacks    rQz /postbackc sTztd|td|ddiWSty)}ztdt|dd}~ww)NzZerodha postback received: %szerodha_postbacksr/receivedr1r)logginginfor get_mongo_db insert_oner2rr#)rBr5r)r)r* kite_postbacks  rXz/profilecCs||ddt|di}|r|dstdddt|d|d |d}z |}d |d WSty=tdd dw) Nrrrr!izNot authenticated with Zerodharrr r-)r/profilez$Invalid/expired Zerodha access token)r%r#r$rr get_profiler2)r&r'r(rOrYr)r)r*get_zerodha_profiles    r[z/webhook/kite/postbackz-Kite Connect postback handler (order updates)c Cr8)NrSr.zkite_postback failedr1r)rrr2r3r4r#)rBresr5r)r)r*rXr7z/stream/feedback/{id}z Stream feedback (thumbs up/down)idc Cs|d}|dvrtdddz4z|ddt|idd |iiWnty9|dd|idd |iiYnwWd d iSWd d iSty\}z td td t|dd}~ww)Nfeedback)updownrJzfeedback must be 'up' or 'down'rstreamrrK user_feedbackzfeedback update failedr1r/r-)r$rrNr r2r3r4r#)r]rBr&fbr5r)r)r*stream_feedbacks$  $   rdz/healthz Health checkcCs&ddlm}|}dt|dS)Nr)validate_openai_configr-)r/timeopenai_configured)app.v1.services.tradeGPTrer utcnow isoformat)re openai_statusr)r)r*healths  rlz /test/llmzTest ChatGPT integrationquestionzIs this working?c CsPddlm}dddidddd d d d d gid}|dd}|||}d|dS)z+Test endpoint to verify ChatGPT integrationr)call_chatgpt_analyze_serviceTEST last_pricegY@5minutez 2023-01-01cebdi)tohlcv)r;quotecandlesrmzAnalyze this test datar-)r/ llm_response)rhrnr$)rBrn mock_snapshotrmr0r)r)r*test_llms    rz/livez=(compat) Live enriched signals - fallback for legacy frontendgainersz,Which top movers to run on: gainers | losers) description2zMax symbols to scanmoverlimitc Csz&t|d}t|||||d}t|tr|dgng}dt||dWSty.tyF}z t dtdt|dd }~ww) z Backwards-compatible endpoint returning rich strategy results. Delegates to get_live_signals_service in services (implement there). r)r&r<rrrresultssuccess)r/scannedrztrade_live failedr1rN) r#r$r isinstancedictlenrr2r3r4) rrr&r'r<rrErr5r)r)r* trade_lives   r)7fastapirrrrrtypingrrr r rTbsonr app.dbr app.v1.dependencies.authr app.v1.services.zerodha.clientrrhrrrrrrrrrouter getLogger__name__r3setLevelINFOrVr+postr6r$r9r#r@rArFrHrrQrXr[rdrlrintrr)r)r)r*s     (                ,   $