Update app.py

app.py

@@ -75,10 +75,19 @@ def answer_question(question, documents):
         "top_p": 0.9  # Control diversity via nucleus sampling
     }
     
+    # Generate the response using the text generation method
     response = client.text_generation(question_with_context, **generate_kwargs)
     
-    # Assuming the response contains a 'generated_text' field with the model's output
-    return response["generated_text"]
+    # Print the response to inspect its structure
+    print(f"Response: {response}")
+    
+    # If the response is a string, we can directly return it
+    if isinstance(response, str):
+        return response
+    else:
+        # If it's a dictionary, extract the generated text
+        return response.get("generated_text", "No answer generated.")
+
 
 
 # Streamlit file uploader for PDF