output clean-up

app.py

@@ -58,7 +58,7 @@ def load_chain():
 chain = load_chain()
 
 # From here down is all the StreamLit UI.
-st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
+st.set_page_config(page_title="Impromptu GPT", page_icon=":robot:")
 st.header("LangChain Demo")
 
 if "generated" not in st.session_state:
@@ -67,27 +67,14 @@ if "generated" not in st.session_state:
 if "past" not in st.session_state:
     st.session_state["past"] = []
 
-
 def get_text():
     input_text = st.text_input("You: ", "What are the four types of hallucinations?", key="input")
     return input_text
 
-
 user_input = get_text()
 
 if user_input:
-    docs = chain.run(user_input)
-    print('docs')
-    print(docs)
-    # print('docs[0]')
-    # print(docs[0])
-    # print('docs[0].page_content')
-    # print(docs[0].page_content)
-
-    # output = docs[0].page_content
-    # output = 'nothing right now'
-    output = docs
-    
+    output = chain.run(user_input)   
     st.session_state.past.append(user_input)
     st.session_state.generated.append(output)