Update app.py

app.py

@@ -39,23 +39,19 @@ def initialize(pdf_file, question):
 
                 prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
 
-                # Generate answer using GeminiPro model
                 # Load the GeminiPro model
                 model = genai.GenerativeModel('gemini-pro')
-                # model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
 
-                # Ensure your LLM object (model) inherits from Runnable and can generate text
+                # Debugging: Print object type and attributes
+                print(type(model))
+                print(dir(model))  # List attributes and methods
 
-                # Prepare the input data
-                input_data = {
-                    "context": processed_context,
-                    "question": question
-                }
+                # Check if the model has a 'generate' method (or similar)
+                if not hasattr(model, 'generate'):
+                    raise Exception("Your LLM object might not have a text generation method!")
 
-                # Generate the answer using load_qa_chain
                 stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
-                # stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
-                stuff_answer = stuff_chain(input_data, return_only_outputs=True)
+                stuff_answer = stuff_chain(input_data={"context": processed_context, "question": question}, return_only_outputs=True)
 
                 # Extract the answer
                 generated_answer = stuff_answer['output_text']