Update app.py

app.py

@@ -124,7 +124,7 @@ def wandb_trace(rag_option, prompt, completion, chain, err_msg, start_time_ms, e
     if (str(err_msg) != ""):
         result = ""
     elif (rag_option == RAG_OFF):
-        result = completion.generations[0][0].text
+        result = completion.generations[0][0].text if (completion.generations[0] != None and completion.generations[0][0] != None) else ""
     else:
         result = completion["result"]
         docs_meta = str([doc.metadata for doc in completion["source_documents"]])
@@ -183,10 +183,7 @@ def invoke(openai_api_key, rag_option, prompt):
             result = completion["result"]
         else:
             completion, chain = llm_chain(llm, prompt)
-            print(completion)
-            print(completion.llm_output)
-            print(completion.generations[0])
-            result = completion.generations[0][0].text
+            result = completion.generations[0][0].text if (completion.generations[0] != None and completion.generations[0][0] != None) else ""
     except Exception as e:
         err_msg = e
         raise gr.Error(e)