Update views/rci.py

views/rci.py

@@ -411,6 +411,18 @@ with stylable_container(
 # Configuração inicial do Groq client
 client = Groq(api_key=st.secrets["GROQ_API_KEY"])
 
+def create_word_doc(markdown_content):
+    # Converter o Markdown em HTML
+    html = markdown.markdown(markdown_content)
+
+    # Converter o HTML em um documento Word
+    document = html2docx(html)
+
+    # Salvar o documento em um objeto BytesIO
+    output = BytesIO()
+    document.save(output)
+    output.seek(0)
+    return output
 
 # Função para gerar a análise inicial
 def generate_initial_analysis(resumo_rci):
@@ -519,16 +531,39 @@ with col1:
         if "initial_analysis" in st.session_state:
             st.write("**Panorama Geral dos Dados:**")
             markdown_content = st.session_state.initial_analysis
-
+        
             html_content = f'''
-            <div style="height:300px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
+            <div style="height:600px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
                 {markdown_content}
             </div>
             '''
-
+        
             st.markdown(html_content, unsafe_allow_html=True)
+        
+            # Botão para baixar o documento em formato Word
+            docx_file = create_word_doc(markdown_content)
+            st.download_button(
+                label="Baixar em formato Word",
+                data=docx_file,
+                file_name="analise_inicial.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )
         else:
             st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
+    
+        # if "initial_analysis" in st.session_state:
+        #     st.write("**Panorama Geral dos Dados:**")
+        #     markdown_content = st.session_state.initial_analysis
+
+        #     html_content = f'''
+        #     <div style="height:500px; overflow-y: auto; padding: 10px; border: 1px solid #CCC;">
+        #         {markdown_content}
+        #     </div>
+        #     '''
+
+        #     st.markdown(html_content, unsafe_allow_html=True)
+        # else:
+        #     st.write("**Clique em 'Gerar Resumo' para analisar os dados.**")
             
         # if "initial_analysis" in st.session_state:
         #     st.write("**Panorama Geral dos Dados:**")