Update app4.py

app4.py

@@ -8,6 +8,7 @@ from langchain_openai import ChatOpenAI
 from langchain.chains import RetrievalQA
 from langchain.schema import Document
 import os
+import re
 
 # Set title
 st.title("Data Analyzer")
@@ -141,6 +142,18 @@ if "df" in st.session_state and api_key and pandasai_api_key:
                     st.write("PandasAI Intermediate Output:", agent.last_output)
             except Exception as e:
                 st.error(f"PandasAI encountered an error: {str(e)}")
+                # Fallback: Direct pandas filtering
+                if "patent_number" in pandas_question.lower() and "decision" in pandas_question.lower():
+                    try:
+                        match = re.search(r'\d{7,}', pandas_question)
+                        if match:
+                            patent_number = match.group()
+                            decision = df.loc[df['patent_number'] == int(patent_number), 'decision']
+                            st.write(f"Fallback Answer: The decision for patent {patent_number} is '{decision.iloc[0]}'.")
+                        else:
+                            st.write("Could not extract patent number from the query.")
+                    except Exception as fallback_error:
+                        st.error(f"Fallback processing failed: {fallback_error}")
 
     with tab2:
         st.subheader("Q&A with RAG")
@@ -158,8 +171,6 @@ if "df" in st.session_state and api_key and pandasai_api_key:
         if viz_question:
             try:
                 result = agent.chat(viz_question)
-                # Extract Python code from PandasAI response
-                import re
                 code_pattern = r'```python\n(.*?)\n```'
                 code_match = re.search(code_pattern, result, re.DOTALL)