Update src/app.py

src/app.py

@@ -11,6 +11,10 @@ from reportlab.lib.pagesizes import A4
 from reportlab.lib import colors
 from reportlab.lib.units import cm
 from reportlab.lib.enums import TA_CENTER, TA_JUSTIFY, TA_LEFT, TA_RIGHT
+from docx import Document
+from docx.shared import Inches
+from io import BytesIO
+from datetime import date
 
 # =================================================================================
 # FUNÇÃO DE GERAÇÃO DE PDF (MODIFICADA PARA ACEITAR DADOS E RETORNAR BYTES)
@@ -160,6 +164,55 @@ def create_pdf_report(user_data):
     buffer.close()
     return pdf_bytes
 
+def create_docx_report(user_data):
+    doc = Document()
+
+    # Título principal
+    doc.add_heading(f"INFORMAÇÃO TÉCNICA - IT_{user_data['it_numero']}", 0)
+
+    doc.add_heading("SOLICITAÇÃO", level=1)
+    doc.add_paragraph(f"Unidade demandante: {user_data['unidade_demandante']}")
+    doc.add_paragraph(f"Finalidade da Avaliação: {user_data['finalidade_avaliacao']}")
+
+    doc.add_heading("IMÓVEL OBJETO", level=1)
+    doc.add_paragraph(f"Endereço - SMF: {user_data['endereco_smf']}")
+    doc.add_paragraph(f"Bairro - SMF: {user_data['bairro_smf']}")
+    doc.add_paragraph(f"Lote Fiscal / Inscrição - SMF: {user_data['lote_fiscal']}")
+    doc.add_paragraph(f"Registro do Imóvel: {user_data['registro_imovel']}")
+    doc.add_paragraph(f"Finalidade Imóvel - SMF:\n{user_data['finalidade_imovel']}")
+    doc.add_paragraph(f"Área Territorial - SMF: {user_data['area_territorial']}")
+    doc.add_paragraph(f"Área construída – SMF:\n{user_data['area_construida']}")
+    doc.add_paragraph(f"Exercícios em análise:\n{user_data['exercicios_analise']}")
+    doc.add_paragraph(f"Valores Venais Guias IPTU (Exercícios): {user_data['valores_venais']}")
+
+    doc.add_heading("ANÁLISE TÉCNICA PRELIMINAR", level=1)
+    doc.add_paragraph(f"Unidade responsável: {user_data['unidade_responsavel']}")
+    doc.add_paragraph(f"Técnico responsável: {user_data['tecnico_responsavel']}")
+    doc.add_paragraph(f"Método de Avaliação:\n{user_data['metodo_avaliacao']}")
+
+    doc.add_heading("CONCLUSÃO TÉCNICA", level=1)
+    doc.add_paragraph(f"Características desvalorizantes: {user_data['caracteristicas_desvalorizantes']}")
+    doc.add_paragraph(f"Conclusão:\n{user_data['conclusao_preliminar']}")
+
+    doc.add_heading("1. CONSIDERAÇÕES INICIAIS", level=1)
+    doc.add_paragraph(user_data['texto_consideracoes'])
+
+    doc.add_heading("2. ANÁLISE TÉCNICA PRELIMINAR", level=1)
+    doc.add_paragraph(user_data['texto_analise'])
+
+    # Data final
+    today = date.today()
+    meses = ("janeiro", "fevereiro", "março", "abril", "maio", "junho",
+             "julho", "agosto", "setembro", "outubro", "novembro", "dezembro")
+    data_formatada = f"Porto Alegre, {today.day} de {meses[today.month - 1]} de {today.year}."
+    doc.add_paragraph("\n" + data_formatada)
+
+    # Salvar o DOCX em memória
+    doc_io = BytesIO()
+    doc.save(doc_io)
+    doc_io.seek(0)
+    return doc_io
+    
 # =================================================================================
 # INTERFACE DO STREAMLIT
 # =================================================================================
@@ -230,4 +283,19 @@ if 'pdf_bytes' in st.session_state and st.session_state.pdf_bytes:
         data=st.session_state.pdf_bytes,
         file_name=st.session_state.file_name,
         mime="application/pdf"
-    )
+    )
+
+if st.button("Gerar Relatório em DOCX"):
+    with st.spinner("Gerando DOCX..."):
+        docx_buffer = create_docx_report(user_data)
+        st.session_state.docx_bytes = docx_buffer.read()
+        st.session_state.docx_file_name = f"IT_{user_data['it_numero'].replace('/', '_')}.docx"
+
+if 'docx_bytes' in st.session_state and st.session_state.docx_bytes:
+    st.success("Seu relatório Word está pronto para download!")
+    st.download_button(
+        label="Baixar DOCX",
+        data=st.session_state.docx_bytes,
+        file_name=st.session_state.docx_file_name,
+        mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+    )