Update app.py

app.py

@@ -8,7 +8,7 @@ import plotly.express as px
 import time
 
 # Title and Sidebar
-st.title("Patent Insights Consultant")
+st.title("🤖 Patent Insights Consultant")
 
 st.sidebar.write(
     "This Patent Insights Consultant uses a multi-agent system to provide actionable insights and data analysis for the patent domain."
@@ -18,6 +18,13 @@ st.sidebar.write(
 patent_area = st.text_input('Enter Patent Technology Area', value="Artificial Intelligence")
 stakeholder = st.text_input('Enter Stakeholder', value="Patent Attorneys")
 
+# Advanced Options
+st.sidebar.subheader("Advanced Options")
+
+# Enable/Disable Advanced Features
+enable_advanced_analysis = st.sidebar.checkbox("Enable Advanced Analysis", value=True)
+enable_custom_visualization = st.sidebar.checkbox("Enable Custom Visualizations", value=True)
+
 # Optional Customization
 st.sidebar.subheader("Agent Customization")
 
@@ -151,6 +158,18 @@ def generate_pdf_report(result):
     pdf.output(report_path)
     return report_path
 
+def create_visualizations(analyst_output):
+    """Create visualizations for advanced insights."""
+    if enable_custom_visualization and analyst_output:
+        try:
+            data = pd.DataFrame(analyst_output)
+            st.markdown("### Advanced Visualization")
+
+            fig = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
+            st.plotly_chart(fig)
+        except Exception as e:
+            st.warning(f"Failed to create visualizations: {e}")
+
 if st.button("Generate Patent Insights"):
     with st.spinner('Processing...'):
         try:
@@ -182,6 +201,10 @@ if st.button("Generate Patent Insights"):
                     analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
                     st.write(analyst_output)
 
+                    # Generate visualizations if enabled
+                    if enable_advanced_analysis:
+                        create_visualizations(analyst_output)
+
             # Display Token Usage and Execution Time
             st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
             token_usage = getattr(results, "token_usage", None)