Update app.py

app.py

@@ -25,7 +25,7 @@ from datasets import load_dataset
 import tempfile
 
 st.title("SQL-RAG Using CrewAI 🚀")
-st.write("Analyze datasets using natural language queries")
+st.write("Analyze datasets using natural language queries powered by SQL and CrewAI.")
 
 # Initialize LLM
 llm = None
@@ -169,15 +169,12 @@ def create_pdf_report_with_viz(report, conclusion, visualizations):
 
     return temp_pdf
 
-
 def escape_markdown(text):
-    # Ensure the input is a string
-    if not isinstance(text, str):
-        text = str(text) if text is not None else ""
-    
-    # Escape special markdown characters
-    escape_chars = r"\*`_{}[]()#+-.!"
-    return re.sub(f'([{re.escape(escape_chars)}])', r'\\\1', 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:
@@ -266,6 +263,8 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
+
+
     # Separate Crews for report and conclusion
     crew_report = Crew(
         agents=[sql_dev, data_analyst, report_writer],
@@ -327,29 +326,8 @@ if st.session_state.df is not None:
                 # Step 6: Display Concise Conclusion
                 #st.markdown("#### Conclusion")
 
-                if conclusion_result:
-                    # Check if conclusion_result is a dictionary or list
-                    if isinstance(conclusion_result, dict):
-                        # Assuming the actual conclusion is inside a 'content' or similar key
-                        conclusion_text = conclusion_result.get('content', '')
-                    elif isinstance(conclusion_result, list):
-                        # If it's a list, join all items into a single string
-                        conclusion_text = " ".join(map(str, conclusion_result))
-                    else:
-                        conclusion_text = str(conclusion_result)
-
-                    # Remove any leading 'Conclusion:' in the result
-                    cleaned_conclusion = re.sub(r'^\s*\*\*?Conclusion:?(\*\*)?\s*', '', conclusion_text, flags=re.IGNORECASE)
-    
-                    # Escape markdown characters
-                    safe_conclusion = escape_markdown(cleaned_conclusion)
-    
-                    # Display the clean conclusion
-                    st.markdown(f"### Conclusion\n\n{safe_conclusion}")
-                else:
-                    st.warning("⚠️ No Conclusion Generated.")
-
-
+                safe_conclusion = escape_markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
+                st.markdown(safe_conclusion)
 
     # Full Data Visualization Tab
     with tab2:
@@ -378,5 +356,3 @@ 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)")
-
-