Update app.py

app.py

@@ -141,7 +141,7 @@ def talk2file(index_name, text):
 
 
     
-    return [context, "Answer:\n" + answer]
+    return [context, answer]
 
 
 
@@ -155,7 +155,7 @@ def fileSearch(index_name, prompt):
     docsearch = Pinecone.from_existing_index(index_name, embeddings)
     docs = docsearch.similarity_search(prompt)
 
-    return "Content in file: \n\n" + docs[0].page_content + "\n\n"
+    return [docs[0].page_content, docs[1].page_content, docs[2].page_content]
 
 
 
@@ -210,6 +210,10 @@ answerbox = gr.inputs.Textbox(label="Assistant answer")
 contextbox = gr.Markdown(label="Context found")
 
 
+contextbox1 = gr.Markdown(label="Top1 context")
+contextbox2 = gr.Markdown(label="Top2 context")
+contextbox3 = gr.Markdown(label="Top3 context")
+
 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=[contextbox, answerbox])
 
@@ -217,10 +221,10 @@ vector_server = gr.Interface(fn=process_file, inputs=["text", gr.inputs.File(fil
 
 #audio = gr.Interface(fn=audioGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs="text")
 #siri = gr.Interface(fn=siriGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs = "audio")
-file = gr.Interface(fn=fileSearch, inputs=["text", "text"], outputs="text", description = "Enter file name and prompt")
+file = gr.Interface(fn=fileSearch, inputs=[textbox, "text"], outputs=[contextbox1, contextbox2, contextbox3], description = "This tab shows the top three most related contexts in the repository.")
 #demo = gr.TabbedInterface([role, text, file, vector_server, pinecone_tools, chatHistory], [ "roleChoice", "Talk2File", "FileSearch", "VectorServer", "PineconeTools", "ChatHistory"])
 
-demo = gr.TabbedInterface([text, file, chatHistory], [ "Talk2File", "FileSearch", "ChatHistory"])
+demo = gr.TabbedInterface([text, file, chatHistory], [ "ROCm Usage Tutor", "Top 3 Context", "ChatHistory"])
 
 if __name__ == "__main__":
     demo.launch(enable_queue=False, auth=lambda u, p: user_db.get(u) == p,