Update app.py

app.py

@@ -25,8 +25,14 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],templat
 def predict(message, history):
     input_prompt = QA_CHAIN_PROMPT.format(question=message, context=history)
     result = llm.generate([input_prompt])
-    print(result)
-    ai_msg = result[0].generations[0].text
+    print(result)  # Print the result for inspection
+    
+    # Access the generated text using the correct attribute(s)
+    if result.generations:
+        ai_msg = result.generations[0][0].text
+    else:
+        ai_msg = "I'm sorry, I couldn't generate a response for that input."
+    
     return ai_msg