Update app.py

app.py

@@ -6,48 +6,58 @@ from fpdf import FPDF
 import pandas as pd
 import plotly.express as px
 import time
-import datetime
-from patent_data_api import fetch_patent_data
 
+# Title and Sidebar
 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."
 )
 
+# User Inputs
 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")
 
-selected_region = st.sidebar.selectbox("Select Region", options=["Global", "United States", "Europe", "China", "Other"])
-start_date = st.sidebar.date_input("Start Date", value=datetime.date(2018, 1, 1))
-end_date = st.sidebar.date_input("End Date", value=datetime.date.today())
-
+# 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)
 
-enable_customization = st.sidebar.checkbox("Enable Custom Goals")
-if enable_customization:
-    planner_goal = st.text_area(
-        "Planner Goal",
-        value="Research trends in patent filings and technological innovation, identify key players, and provide strategic recommendations."
-    )
-    writer_goal = st.text_area(
-        "Writer Goal",
-        value="Craft a professional insights document summarizing trends, strategies, and actionable outcomes for stakeholders."
-    )
-    analyst_goal = st.text_area(
-        "Analyst Goal",
-        value="Perform detailed statistical analysis of patent filings, growth trends, and innovation distribution."
-    )
-else:
-    planner_goal = "Research trends in patent filings and technological innovation, identify key players, and provide strategic recommendations."
-    writer_goal = "Craft a professional insights document summarizing trends, strategies, and actionable outcomes for stakeholders."
-    analyst_goal = "Perform detailed statistical analysis of patent filings, growth trends, and innovation distribution."
-
+# Optional Customization
+st.sidebar.subheader("Agent Customization")
+
+# Display customization section in a collapsible expander
+with st.sidebar.expander("Customize Agent Goals", expanded=False):
+    enable_customization = st.checkbox("Enable Custom Goals")
+    if enable_customization:
+        planner_goal = st.text_area(
+            "Planner Goal",
+            value="Research trends in patent filings and technological innovation, identify key players, and provide strategic recommendations."
+        )
+        writer_goal = st.text_area(
+            "Writer Goal",
+            value="Craft a professional insights document summarizing trends, strategies, and actionable outcomes for stakeholders."
+        )
+        analyst_goal = st.text_area(
+            "Analyst Goal",
+            value="Perform detailed statistical analysis of patent filings, growth trends, and innovation distribution."
+        )
+    else:
+        planner_goal = "Research trends in patent filings and technological innovation, identify key players, and provide strategic recommendations."
+        writer_goal = "Craft a professional insights document summarizing trends, strategies, and actionable outcomes for stakeholders."
+        analyst_goal = "Perform detailed statistical analysis of patent filings, growth trends, and innovation distribution."
+
+#=================
+# LLM Object
+#=================
 llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model="groq/llama-3.3-70b-versatile")
 
+#=================
+# Crew Agents
+#=================
+
 planner = Agent(
     role="Patent Research Consultant",
     goal=planner_goal,
@@ -81,6 +91,10 @@ analyst = Agent(
     llm=llm
 )
 
+#=================
+# Crew Tasks
+#=================
+
 plan = Task(
     description=(
         "1. Research recent trends in {topic} patent filings and innovation.\n"
@@ -113,6 +127,10 @@ analyse = Task(
     agent=analyst
 )
 
+#=================
+# Execution
+#=================
+
 crew = Crew(
     agents=[planner, analyst, writer],
     tasks=[plan, analyse, write],
@@ -120,29 +138,34 @@ crew = Crew(
 )
 
 def generate_pdf_report(result):
+    """Generate a professional PDF report from the Crew output."""
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", size=12)
     pdf.set_auto_page_break(auto=True, margin=15)
 
+    # Title
     pdf.set_font("Arial", size=16, style="B")
     pdf.cell(200, 10, txt="Patent Insights Report", ln=True, align="C")
     pdf.ln(10)
 
+    # Content
     pdf.set_font("Arial", size=12)
     pdf.multi_cell(0, 10, txt=result)
 
+    # Save PDF
     report_path = "Patent_Insights_Report.pdf"
     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.scatter(data, x="Category", y="Values", title="Innovation Trends by Category", size="Impact", color="Region")
+            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}")
@@ -151,14 +174,10 @@ if st.button("Generate Patent Insights"):
     with st.spinner('Processing...'):
         try:
             start_time = time.time()
-
-            real_time_data = fetch_patent_data(
-                technology_area=patent_area, region=selected_region, start_date=start_date, end_date=end_date
-            )
-
-            results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder, "data": real_time_data})
+            results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
             elapsed_time = time.time() - start_time
 
+            # Display Final Report (Writer's Output)
             st.markdown("### Final Report:")
             writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
             if writer_output:
@@ -166,28 +185,34 @@ if st.button("Generate Patent Insights"):
             else:
                 st.warning("No final report available.")
 
+            # Option for Detailed Insights
             with st.expander("Explore Detailed Insights"):
                 tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
 
+                # Planner's Output
                 with tab1:
                     st.markdown("### Planner's Insights")
                     planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
                     st.write(planner_output)
 
+                # Analyst's Output
                 with tab2:
                     st.markdown("### Analyst's Analysis")
                     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)
             if token_usage:
                 st.markdown("#### Token Usage")
                 st.json(token_usage)
 
+            # Generate PDF Report
             if writer_output:
                 report_path = generate_pdf_report(writer_output)
                 with open(report_path, "rb") as report_file:
@@ -196,5 +221,6 @@ if st.button("Generate Patent Insights"):
         except Exception as e:
             st.error(f"An error occurred during execution: {e}")
 
+# Add reference and credits in the sidebar
 st.sidebar.markdown("---")
 st.sidebar.markdown("### Reference:")