Update app.py

app.py

@@ -164,6 +164,7 @@ def invoke(openai_api_key, rag_option, prompt):
     chain = None
     completion = ""
     result = ""
+    generation_info = ""
     llm_output = ""
     err_msg = ""
     try:
@@ -187,11 +188,13 @@ def invoke(openai_api_key, rag_option, prompt):
             completion, chain = llm_chain(llm, prompt)
             if (completion.generations[0] != None and completion.generations[0][0] != None):
                 result = completion.generations[0][0].text
-                generation_info = completion.generations[0][0].generation_info
-                llm_output = completion.generations.llm_output
+                print("###")
                 print(completion)
                 print("###")
                 print(completion.generations[0])
+                print("###")
+                generation_info = completion.generations[0][0].generation_info
+                #llm_output = completion.generations.llm_output
     except Exception as e:
         err_msg = e
         raise gr.Error(e)