o Bi] @sJddlZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z e jZd&d ed efd d Zd'd edefddZedkrejddZejdddZejdddZejddddejdedddejd d!dZejddd"dejd#d d$d%eZejdkreejej dSejd kreejej!dSdSdS)(N)Path) FaissStore) embed_queryanswer_question_with_llm)build_report_from_query) render_docx)configquerykc Cstd|dtt}t|}|j||d}|s tddStdt|dD]D\}}|d}|d d }td |d |d dd|ddd|dtd|ddtd|ddddtdq)tdt||} td| ddS)zS Q&A mode: retrieve relevant chunks and optionally generate an LLM answer. u 🔍 Question:  )r u3❌ No relevant information found in the documents.Nu📄 Retrieved Evidence: metadatascoreg[z] doc_nameUnknownz | Page page_number?z | Score: z.3fz Chunk ID: chunk_idzN/Az Text: texti,z... zP--------------------------------------------------------------------------------u 🤖 Generating Answer... u ✅ Answer: ) printrload INDEX_DIRrsearch enumerategetr) r r storeqvecresultsidxrmetaranswerr$:/var/www/html/fyndo/python/python_agents/rag_suite/main.py ask_commands$  2  r&outputs/reports/report.docxout_filecCsVtd|dtt}t||}t|jjdddt||}td|ddS)z_ Report mode: retrieve relevant chunks, generate structured sections, and render DOCX. u 📝 Generating Report for: r T)parentsexist_oku' ✅ Report generated successfully at: N) rrrrrrparentmkdirr)r r(r report_obj final_pathr$r$r%report_command+s    r/__main__zCRAG Suite CLI - Ask questions or generate reports from indexed PDFs) descriptioncommandT)destrequiredaskzAsk a question (Q&A mode))helpz--qz Your question)r4r6z--kz)Number of chunks to retrieve (default: 8))typedefaultr6reportzGenerate a structured reportzReport query/promptz--outzOutput path for DOCX)r8r6)r )r')"argparsepathlibrrag.vectorstorerrag.qarr rag.reportr rag.renderrragrrstrintr&r/__name__ArgumentParserparseradd_subparsers subparsers add_parser parser_ask add_argument parser_report parse_argsargsr2qr outr$r$r$r%s8