Update app.py

app.py

@@ -49,19 +49,8 @@ def generateResponse(prompt, dfs):
     llm = GoogleGemini(api_key=GOOGLE_API_KEY)
     response_parser = StreamLitResponse({})
     pandas_agent = SmartDatalake(dfs, config={"llm": llm, "response_parser": response_parser})
-    try:
-        
-        results = pandas_agent.chat(prompt)
-        if not results:  # If results is empty
-            return {"type": "text", "value": "No results found for your query."}  # Return a default message as a dictionary
-        elif len(results) == 1:  # If results contains only one object, return the object directly
-            return results[0]
-        else:
-            return results  # Return the list of results
-    except Exception as e:  # Catch any exceptions during PandasAI processing
-        st.error(f"Error in PandasAI: {e}")  # Display the error in Streamlit
-        return {"type": "text", "value": f"An error occurred: {e}"} 
-
+    answer = pandas_agent.chat(prompt)
+    return answer
 
 # Processing pdfs
 def get_pdf_text(pdf_docs):