Update app.py

app.py

@@ -20,7 +20,7 @@ from langchain.document_loaders import DirectoryLoader
 from langchain.indexes import VectorstoreIndexCreator
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Pinecone
-import markdown2
+import markdown
 
 
 openai.api_key = os.environ['OPENAI_API_KEY']
@@ -133,8 +133,8 @@ def talk2file(index_name, text):
 
     #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)
+    Title2 = '<h2"><b>Context Found: </b></h2>'
+    context = markdown.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 = system_message["content"]
@@ -206,8 +206,9 @@ with gr.Blocks() as pinecone_tools:
 
 
 textbox = gr.inputs.Textbox(label="Vector Server Index Name: ", default="amd")
+answerbox = gr.inputs.Textbox(label="Assistant answer"")
 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"])
+text = gr.Interface(fn=talk2file, inputs=[textbox, "text"], outputs=["html", answerbox])
 
 vector_server = gr.Interface(fn=process_file, inputs=["text", gr.inputs.File(file_count="directory")], outputs="text")