Update app.py

app.py

@@ -70,3 +70,93 @@ elif input_option == "Upload CSV File":
         st.session_state.df = pd.read_csv(uploaded_file)
         st.success("File uploaded successfully!")
         st.dataframe(st.session_state.df.head())
+
+
+if st.session_state.df is not None:
+    # Database setup
+    temp_dir = tempfile.TemporaryDirectory()
+    db_path = os.path.join(temp_dir.name, "patent_data.db")
+    connection = sqlite3.connect(db_path)
+    st.session_state.df.to_sql("patents", connection, if_exists="replace", index=False)
+    db = SQLDatabase.from_uri(f"sqlite:///{db_path}")
+
+    # SQL Tools
+    @tool("list_tables")
+    def list_tables() -> str:
+        """List all tables in the patent database."""
+        return ListSQLDatabaseTool(db=db).invoke("")
+
+    @tool("tables_schema")
+    def tables_schema(tables: str) -> str:
+        """Get schema and sample rows for given tables."""
+        return InfoSQLDatabaseTool(db=db).invoke(tables)
+
+    @tool("execute_sql")
+    def execute_sql(sql_query: str) -> str:
+        """Execute a SQL query against the patent database."""
+        return QuerySQLDataBaseTool(db=db).invoke(sql_query)
+
+    # --- CrewAI Agents for Patent Analysis ---
+    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.",
+        llm=llm,
+        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.",
+        llm=llm,
+    )
+
+    report_writer = Agent(
+        role="Patent Report Writer",
+        goal="Summarize patent insights into a clear report.",
+        backstory="Expert in summarizing patent data into comprehensive reports.",
+        llm=llm,
+    )
+
+    # --- Crew Tasks ---
+    extract_data = Task(
+        description="Extract patents related to the query: {query}.",
+        expected_output="Patent data matching the query.",
+        agent=sql_dev,
+    )
+
+    analyze_data = Task(
+        description="Analyze the extracted patent data.",
+        expected_output="Detailed analysis of patents.",
+        agent=data_analyst,
+        context=[extract_data],
+    )
+
+    write_report = Task(
+        description="Summarize analysis into an executive report.",
+        expected_output="Markdown report of patent insights.",
+        agent=report_writer,
+        context=[analyze_data],
+    )
+
+    # Assemble Crew
+    crew = Crew(
+        agents=[sql_dev, data_analyst, 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.'")
+    if st.button("Submit Query"):
+        with st.spinner("Processing your query..."):
+            inputs = {"query": query}
+            result = crew.kickoff(inputs=inputs)
+            st.markdown("### 📊 Patent Analysis Report")
+            st.markdown(result)
+
+    temp_dir.cleanup()
+else:
+    st.info("Please load a patent dataset to proceed.")