Update app.py

app.py

@@ -7,17 +7,17 @@ from crewai_tools import PDFSearchTool, FileReadTool, DOCXSearchTool, CSVSearchT
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_csv_agent
-from langchain_groq import ChatGroq
 
 # API keys-----------------move them to ENV 
 os.environ["OPENAI_API_KEY"] = "NA"
 os.environ["GOOGLE_API_KEY"] = "AIzaSyD7jKc5MdkRLakxcyhvrpie8XgbwY98NMo"
-os.environ["GROQ_API_KEY"]='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'
 
-# Load The Groq model for LLM
-llm = ChatGroq(
-    api_key="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",
-    model="llama3-70b-8192"
+# Load The Gemini model for LLM
+llm = ChatGoogleGenerativeAI(
+    model="gemini-1.5-flash-latest",
+    verbose=True,               
+    temperature=0.6,            # high temp=high accuracy and low creativity                                   
+    google_api_key="AIzaSyD7jKc5MdkRLakxcyhvrpie8XgbwY98NMo"
 )
 
 #<-----------------------------Tools----------------------------------->
@@ -26,9 +26,9 @@ class tools:
         PDFtool = PDFSearchTool(
             config=dict(
                 llm=dict(
-                    provider="ollama",
+                    provider="google",
                     config=dict(
-                        model="llama3-70b-8192",
+                        model="gemini-1.5-flash-latest",
                     ),
                 ),
                 embedder=dict(
@@ -47,9 +47,9 @@ class tools:
         Filetool = FileReadTool(
             config=dict(
                 llm=dict(
-                    provider="ollama",
+                    provider="google",
                     config=dict(
-                        model="llama3-70b-8192",
+                        model="gemini-1.5-flash-latest",
                     ),
                 ),
                 embedder=dict(
@@ -68,9 +68,9 @@ class tools:
         Docstool = DOCXSearchTool(
             config=dict(
                 llm=dict(
-                    provider="ollama",
+                    provider="google",
                     config=dict(
-                        model="llama3-70b-8192",
+                        model="gemini-1.5-flash-latest",
                     ),
                 ),
                 embedder=dict(
@@ -92,7 +92,7 @@ class AgentLoader:
 
     def csvReaderAgent(path):
         agent = create_csv_agent(
-           ChatGroq(temperature=0.6, model="llama3-70b-8192"),
+            ChatGoogleGenerativeAI(temperature=0.6, model="gemini-1.5-flash-latest"),
             path,
             verbose=True,
             agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION
@@ -144,13 +144,11 @@ class AgentLoader:
     def writerAgent():
         writer=Agent(
             role='Content Writer',
-            goal='To provide QUICK and reliable output',
-            backstory="""You are content specialist.
-            You are responsible to generate high quality results in the required format very quickly as soon as data is available.
-            You are very accurate and fast at the same time.""",
+            goal='To produce higly accurate and easy to understand information',
+            backstory="""You are an content specialist and are respinsible to generate reliable and easy to understand content or information based on the summary of data.
+            You should provide indetail results on the summary data.""",
             verbose=True,
-            llm=llm,
-            max_iter=5
+            llm=llm
         )
         return writer
 
@@ -213,4 +211,4 @@ interface = gr.Interface(
 )
 
 # Launch the Gradio interface
-interface.launch()
+interface.launch()