Update app.py

app.py

@@ -2,16 +2,32 @@
 from gpt_index import SimpleDirectoryReader, GPTListIndex, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
 from langchain import OpenAI
 import gradio as gr
+import random
+import time
 import sys
 import os
+from transformers import pipeline
+p = pipeline("automatic-speech-recognition")
+
 
 os.environ["OPENAI_API_KEY"] = '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'
 
+md = """This is some code:
+
+hello
+
+```py
+def fn(x, y, z):
+    print(x, y, z)
+"""
+def transcribe(audio):
+    text = p(audio)["text"]
+    return text
 def construct_index(directory_path):
-    max_input_size = 4096
-    num_outputs = 512
-    max_chunk_overlap = 20
-    chunk_size_limit = 600
+    max_input_size = 10000
+    num_outputs = 10000
+    max_chunk_overlap = 20000
+    chunk_size_limit = 600000
 
     prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
 
@@ -26,22 +42,33 @@ def construct_index(directory_path):
 
 
 def chatbot(input_text):
+
     index = GPTSimpleVectorIndex.load_from_disk('index.json')
     response = index.query(input_text)
-    return response.response
+    return str(response.response)
 
-description = """
 
-<center>Olá sou a Zoh, fui treinada para responder perguntas com base das informações do Hippo Supermercados. Pergunte qualquer coisa. Caso eu não saiba, estarei aprendendo.
-<img src="https://s3.amazonaws.com/enlizt-resources-prod/companies/10958750-6306-11ea-b31c-2b332181af51_256_avatar?nocache=1588599205314" width=200px></center>
-"""
+with gr.Blocks() as demo:
+    gpt = gr.Chatbot(label="GPT SUPEr", elem_id="chatbot").style(height=800)
+    msg = gr.Textbox( show_label=False,
+                placeholder="Bem vindo ao ExpoSuper, Qual sua pergunta?",
+            ).style(container=False)
+    clear = gr.Button("Limpar Conversa")
+    gr.Audio(source="microphone", type="filepath",label="ESTÁ COM DIFICULDADES EM ESCREVER? CLIQUE E ME DIGA O QUE DESEJA")
+    def respond(message, chat_history):
+        chat_history.append((message, chatbot(message)))
+        time.sleep(1)
+        vetor = []
+        realPath = str(os.path.dirname(os.path.realpath(__file__)))
 
-iface = gr.Interface(fn=chatbot,
-                     inputs=gr.inputs.Textbox(lines=3, label='O quê gostaria de saber?') ,
-                     outputs=gr.inputs.Textbox(lines=3, label="Veja o que encontrei"),
-                     description=description,
-                     css=".gradio-container-3-23-0 {background-color: #5f0000} .gradio-container-3-23-0 .prose * {color: #ffffff}",
-                     title="CD2 IA")
+        if str(message).upper()=="OLA" or str(message).upper()=="OLÁ" or str(message).upper()=="OI":
+            vetor = vetor + [((realPath + "\\images\\logo.png",), "")]
+        
+        return "", chat_history+vetor
+
+    clear.click(lambda:None, None, gpt, queue=False,)
+    msg.submit(respond, [msg, gpt], [msg,gpt])
 
 index = construct_index("docs")
-iface.launch()
+demo.launch()
+