Update app.py

app.py

@@ -20,6 +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
 
 
 openai.api_key = os.environ['OPENAI_API_KEY']
@@ -104,8 +105,10 @@ def roleChoice(role):
 
 
 
-def talk2file(index_name, text):
-    global messages
+def talk2file(index_name=amd, text):
+    #disable the global message
+    #global messages
+    messages = []
     
     #same as filesearch
     init_pinecone()
@@ -114,19 +117,29 @@ def talk2file(index_name, text):
     docs = docsearch.similarity_search(text)
 
     
-    prompt = text + ", based on the following context: \n\n" + docs[0].page_content 
-    messages.append({"role": "user", "content": prompt})
+    prompt = text + ", based on the following context: \n\n" 
+    qwcontext = prompt + docs[0].page_content
+    messages.append({"role": "user", "content": qwcontext})
 
     response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
 
     system_message = response["choices"][0]["message"]
-    messages.append(system_message)
+    #messages.append(system_message)
 
-    chats = ""
-    for msg in messages:
-        if msg['role'] != 'system':
-            chats += msg['role'] + ": " + msg['content'] + "\n\n"
+    #chats = ""
+    #for msg in messages:
+    #    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'<span style="background-color: yellow;">{prompt}</span>'
+    Title2 = '<h2 style="background-color: blue;"><b>Context Found: </b></h2>'
+    context = f'<span style="background-color: blue;">{docs[0].page_content}</span>'
+    Title3 = '<h2 style="background-color: green;"><b>Ansewr: </b></h2>'
+    User_Question = f'<span style="background-color: green;">{system_message}</span>'
 
+
+    
     return chats
 
 
@@ -193,7 +206,7 @@ with gr.Blocks() as pinecone_tools:
 
 
 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=["text", "text"], outputs="text")
+text = gr.Interface(fn=talk2file, inputs=["text", "text"], outputs="html")
 
 vector_server = gr.Interface(fn=process_file, inputs=["text", gr.inputs.File(file_count="directory")], outputs="text")