Update interim.py

interim.py

@@ -1,7 +1,10 @@
 import streamlit as st
 import pandas as pd
 import sqlite3
+import tempfile
+from fpdf import FPDF
 import os
+import re
 import json
 from pathlib import Path
 import plotly.express as px
@@ -83,6 +86,96 @@ if st.session_state.df is not None and st.session_state.show_preview:
     st.subheader("📂 Dataset Preview")
     st.dataframe(st.session_state.df.head())
 
+# Function to create TXT file
+def create_text_report_with_viz_temp(report, conclusion, visualizations):
+    content = f"### Analysis Report\n\n{report}\n\n### Visualizations\n"
+
+    for i, fig in enumerate(visualizations, start=1):
+        fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
+        x_axis = fig.layout.xaxis.title.text if fig.layout.xaxis.title.text else "X-axis"
+        y_axis = fig.layout.yaxis.title.text if fig.layout.yaxis.title.text else "Y-axis"
+
+        content += f"\n{i}. {fig_title}\n"
+        content += f"   - X-axis: {x_axis}\n"
+        content += f"   - Y-axis: {y_axis}\n"
+
+        if fig.data:
+            trace_types = set(trace.type for trace in fig.data)
+            content += f"   - Chart Type(s): {', '.join(trace_types)}\n"
+        else:
+            content += "   - No data available in this visualization.\n"
+
+    content += f"\n\n\n{conclusion}"
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w', encoding='utf-8') as temp_txt:
+        temp_txt.write(content)
+        return temp_txt.name
+
+
+# Function to create PDF with report text and visualizations
+def create_pdf_report_with_viz(report, conclusion, visualizations):
+    pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+
+    # Title
+    pdf.set_font("Arial", style="B", size=18)
+    pdf.cell(0, 10, "📊 Analysis Report", ln=True, align="C")
+    pdf.ln(10)
+
+    # Report Content
+    pdf.set_font("Arial", style="B", size=14)
+    pdf.cell(0, 10, "Analysis", ln=True)
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, report)
+
+    pdf.ln(10)
+    pdf.set_font("Arial", style="B", size=14)
+    pdf.cell(0, 10, "Conclusion", ln=True)
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, conclusion)
+
+    # Add Visualizations
+    pdf.add_page()
+    pdf.set_font("Arial", style="B", size=16)
+    pdf.cell(0, 10, "📈 Visualizations", ln=True)
+    pdf.ln(5)
+
+    with tempfile.TemporaryDirectory() as temp_dir:
+        for i, fig in enumerate(visualizations, start=1):
+            fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
+            x_axis = fig.layout.xaxis.title.text if fig.layout.xaxis.title.text else "X-axis"
+            y_axis = fig.layout.yaxis.title.text if fig.layout.yaxis.title.text else "Y-axis"
+
+            # Save each visualization as a PNG image
+            img_path = os.path.join(temp_dir, f"viz_{i}.png")
+            fig.write_image(img_path)
+
+            # Insert Title and Description
+            pdf.set_font("Arial", style="B", size=14)
+            pdf.multi_cell(0, 10, f"{i}. {fig_title}")
+            pdf.set_font("Arial", size=12)
+            pdf.multi_cell(0, 10, f"X-axis: {x_axis} | Y-axis: {y_axis}")
+            pdf.ln(3)
+
+            # Embed Visualization
+            pdf.image(img_path, w=170)
+            pdf.ln(10)
+
+        # Save PDF
+        temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+        pdf.output(temp_pdf.name)
+
+    return temp_pdf
+
+def escape_markdown(text):
+    # Ensure text is a string
+    text = str(text)
+    # Escape Markdown characters: *, _, `, ~
+    escape_chars = r"(\*|_|`|~)"
+    return re.sub(escape_chars, r"\\\1", text)
+
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()
@@ -129,7 +222,7 @@ if st.session_state.df is not None:
 
     report_writer = Agent(
         role="Technical Report Writer",
-        goal="Write a structured report with Key Insights and Analysis. DO NOT include Conclusion or Summary.",
+        goal="Write a structured report with Introduction and Key Insights. DO NOT include any Conclusion or Summary.",
         backstory="Specializes in detailed analytical reports without conclusions.",
         llm=llm,
     )
@@ -156,19 +249,22 @@ if st.session_state.df is not None:
     )
 
     write_report = Task(
-        description="Write the analysis report with Key Insights. DO NOT include a Conclusion.",
+        description="Write the analysis report with  Introduction and Key Insights. DO NOT include any Conclusion or Summary.",
         expected_output="Markdown-formatted report excluding Conclusion.",
         agent=report_writer,
         context=[analyze_data],
     )
 
     write_conclusion = Task(
-        description="Write a brief and impactful 3-5 line Conclusion summarizing only the most important insights.",
-        expected_output="Markdown-formatted concise Conclusion section.",
+        description="Summarize the key findings in 3-5 impactful lines, highlighting the maximum, minimum, and average salaries."
+        "Emphasize significant insights on salary distribution and influential compensation trends for strategic decision-making.",
+        expected_output="Markdown-formatted Conclusion section with key insights and statistics.",
         agent=conclusion_writer,
         context=[analyze_data],
     )
 
+
+
     # Separate Crews for report and conclusion
     crew_report = Crew(
         agents=[sql_dev, data_analyst, report_writer],
@@ -201,7 +297,7 @@ if st.session_state.df is not None:
                 conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
 
                 # Step 3: Display the report
-                st.markdown("### Analysis Report:")
+                #st.markdown("### Analysis Report:")
                 st.markdown(report_result if report_result else "⚠️ No Report Generated.")
 
                 # Step 4: Generate Visualizations
@@ -223,13 +319,15 @@ if st.session_state.df is not None:
                 visualizations.append(fig_employment)
 
                 # Step 5: Insert Visual Insights
-                st.markdown("## 📊 Visual Insights")
+                st.markdown("### Visual Insights")
                 for fig in visualizations:
                     st.plotly_chart(fig, use_container_width=True)
 
                 # Step 6: Display Concise Conclusion
-                st.markdown("## Conclusion")
-                st.markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
+                #st.markdown("#### Conclusion")
+
+                safe_conclusion = escape_markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
+                st.markdown(safe_conclusion)
 
     # Full Data Visualization Tab
     with tab2:
@@ -258,3 +356,4 @@ else:
 with st.sidebar:
     st.header("📚 Reference:")
     st.markdown("[SQL Agents w CrewAI & Llama 3 - Plaban Nayak](https://github.com/plaban1981/Agents/blob/main/SQL_Agents_with_CrewAI_and_Llama_3.ipynb)")
+