Update app.py

app.py

@@ -68,10 +68,9 @@ texts = text_splitter.split_documents(docs)
 embedding_model = Embedding(model_name="BAAI/bge-base-en", max_length=512)
 embeddings = list(embedding_model.embed([text.page_content for text in texts]))
 
-print(embeddings)
 
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(height=700)
     msg = gr.Textbox()
     clear = gr.ClearButton([msg, chatbot])
 
@@ -101,13 +100,11 @@ with gr.Blocks() as demo:
         bot_message = ""
         
         if output[0]["generated_text"]:
-            bot_message = f"""{output[0]["generated_text"]}
-            
-        Sources
+            bot_message = output[0]["generated_text"]
+            bot_message += "\n \n"
         
-        """
-        for i, doc in enumerate(result_docs):
-            bot_message += f"⚫️ Source {i+1} ⚫️ {doc.page_content}\n"
+            for i, doc in enumerate(result_docs):
+                bot_message += f"⚫️ Source: {i+1} {doc.page_content}\n"
  
         chat_history.append((message, bot_message))
         return "", chat_history