Update app.py

app.py

@@ -167,11 +167,13 @@ def create_vector_db(final_items):
         Once the summarizing is done, write 'END'.
         """
         
-        response = llm.generate(
+        response2 = llm.generate(
             prompt
         )
 
-        documents.append(response["choices"][0]["text"].strip())
+        response = response2["choices"][0]["text"].strip()
+
+        documents.append(response)
     
     if final_items:
         db.add(
@@ -215,11 +217,13 @@ def generate_response(db, query_text, previous_context):
 
     Once you are done summarizing, type 'END'.
     """
-    response = llm(
+    response2 = llm(
         prompt_template
     )
     
-    print(response["choices"][0]["text"].strip())
+    response = response2["choices"][0]["text"].strip()
+
+    print(response)
     
     
 def streamlit_app():