Update app.py

app.py

@@ -97,7 +97,7 @@ if st.session_state.df is not None:
         return QuerySQLDataBaseTool(db=db).invoke(sql_query)
 
     # --- CrewAI Agents for Patent Analysis ---
-    sql_dev = Agent(
+    patent_sql_dev = Agent(
         role="Patent Data Analyst",
         goal="Extract patent data using optimized SQL queries.",
         backstory="An expert in writing optimized SQL queries for complex patent databases.",
@@ -105,17 +105,17 @@ if st.session_state.df is not None:
         tools=[list_tables, tables_schema, execute_sql],
     )
 
-    data_analyst = Agent(
-        role="Patent Analyst",
-        goal="Analyze patent claims and decisions for insights.",
-        backstory="Experienced in analyzing patent trends and risk factors.",
+    patent_data_analyst = Agent(
+        role="Patent Data Analyst",
+        goal="Analyze the data and produce insights.",
+        backstory="A seasoned analyst who identifies trends and patterns in datasets.",
         llm=llm,
     )
 
-    report_writer = Agent(
+    patent_report_writer = Agent(
         role="Patent Report Writer",
         goal="Summarize patent insights into a clear report.",
-        backstory="Expert in summarizing patent data into comprehensive reports.",
+        backstory="Expert in summarizing patent data insights into comprehensive reports.",
         llm=llm,
     )
 
@@ -123,33 +123,33 @@ if st.session_state.df is not None:
     extract_data = Task(
         description="Extract patents related to the query: {query}.",
         expected_output="Patent data matching the query.",
-        agent=sql_dev,
+        agent=patent_sql_dev,
     )
 
     analyze_data = Task(
-        description="Analyze the extracted patent data.",
-        expected_output="Detailed analysis of patents.",
-        agent=data_analyst,
+        description="Analyze the extracted patent data for query: {query}.",
+        expected_output="Analysis text summarizing findings.",
+        agent=patent_data_analyst,
         context=[extract_data],
     )
 
     write_report = Task(
         description="Summarize analysis into an executive report.",
-        expected_output="Markdown report of patent insights.",
+        expected_output="Markdown report of insights.",
         agent=report_writer,
-        context=[analyze_data],
+        context=[patent_analyze_data],
     )
 
     # Assemble Crew
     crew = Crew(
-        agents=[sql_dev, data_analyst, report_writer],
+        agents=[patent_sql_dev, patent_data_analyst, patent_report_writer],
         tasks=[extract_data, analyze_data, write_report],
         process=Process.sequential,
         verbose=True,
     )
 
     #Query Input for Patent Analysis
-    query = st.text_area("Enter Patent Analysis Query:", placeholder="e.g., 'List AI patents filed after 2018.'")
+    query = st.text_area("Enter Patent Analysis Query:", placeholder="e.g., 'How many patents related to Machine Learning were filed after 2016?'")
     if st.button("Submit Query"):
         with st.spinner("Processing your query..."):
             inputs = {"query": query}