Update app.py

app.py

@@ -10,6 +10,8 @@ import PyPDF2
 import os
 import json
 from dotenv import load_dotenv
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
 
 # Load environment variables
 load_dotenv()
@@ -44,12 +46,21 @@ class SmartShoppingAssistant:
             )
         ]
         
-        prompt = "You are a shopping assistant. Use the tools provided to find products based on user queries."
+        # Define a prompt template for the agent
+        prompt_template = """
+        You are a shopping assistant. Use the tools provided to find products based on user queries.
+        {input}
+        """
         
+        # Create a PromptTemplate and LLMChain to make the model compatible
+        prompt = PromptTemplate(input_variables=["input"], template=prompt_template)
+        llm_chain = LLMChain(llm=llm_flash_exp, prompt=prompt)
+
+        # Create the agent with the adapted chain
         agent = create_openai_functions_agent(
-            llm=llm_flash_exp,
+            llm=llm_chain,
             tools=tools,
-            prompt=prompt
+            prompt=prompt_template
         )
         
         self.agent_executor = AgentExecutor(
@@ -85,21 +96,6 @@ class SmartShoppingAssistant:
             return "\n\n".join(results) if results else "No matching products found."
         except Exception as e:
             return f"Error processing query: {str(e)}"
-    
-    def extract_text_from_image(self, image):
-        try:
-            response = model_flash_think.generate_content(image)
-            return response.text
-        except Exception as e:
-            return f"Error processing image: {str(e)}"
-    
-    def extract_text_from_pdf(self, pdf_file):
-        try:
-            pdf_reader = PyPDF2.PdfReader(pdf_file)
-            text = "".join([page.extract_text() for page in pdf_reader.pages])
-            return text
-        except Exception as e:
-            return f"Error processing PDF: {str(e)}"
 
 def main():
     st.set_page_config(page_title="Smart Shopping Assistant", layout="wide")