Update app.py

app.py

@@ -131,17 +131,17 @@ def talk2file(index_name, text):
     #    if msg['role'] != 'system':
     #        chats += msg['role'] + ": " + msg['content'] + "\n\n"
 
-    Title1 = '<h2 style="background-color: yellow;"><b>User Question: </b></h2>'
-    User_Question = f'<div style="background-color: yellow; display: inline-block; word-wrap: break-word;">{prompt}</div>'
+    #Title1 = '<h2 style="background-color: yellow;"><b>User Question: </b></h2>'
+    #User_Question = f'<div style="background-color: yellow; display: inline-block; word-wrap: break-word;">{prompt}</div>'
     Title2 = '<h2 style="background-color: #ADD8E6;"><b>Context Found: </b></h2>'
     context = markdown2.markdown(docs[0].page_content)
     #context = f'<span style="background-color: #ADD8E6; display: inline-block; word-wrap: break-word;">{context2html}</span>'
-    Title3 = '<h2 style="background-color: #90EE90;"><b>Ansewr: </b></h2>'
-    answer = f'<div style="background-color: #90EE90; display: inline-block; overflow-wrap: break-word; word-break: break-all;">{system_message["content"]}</div>'
+    #Title3 = '<h2 style="background-color: #90EE90;"><b>Ansewr: </b></h2>'
+    answer = system_message["content"]
 
 
     
-    return Title1 + User_Question + Title2 + context + Title3 + answer
+    return [Title2 + context, "Answer:\n" + answer]
 
 
 
@@ -207,7 +207,7 @@ with gr.Blocks() as pinecone_tools:
 
 textbox = gr.inputs.Textbox(label="Vector Server Index Name: ", default="amd")
 role = gr.Interface(fn=roleChoice, inputs="text", outputs="text", description = "Choose your GPT roles, e.g. You are a helpful technology assistant.")
-text = gr.Interface(fn=talk2file, inputs=[textbox, "text"], outputs="html")
+text = gr.Interface(fn=talk2file, inputs=[textbox, "text"], outputs=["html", "text"])
 
 vector_server = gr.Interface(fn=process_file, inputs=["text", gr.inputs.File(file_count="directory")], outputs="text")