Update app.py

app.py

@@ -30,10 +30,24 @@ def chat(message, site,history):
     response = ""
     try:
         response = chain.run(input=message)
-    history.append((message, response))
+        markdown = generate_markdown(response)
+    history.append((message, markdown))
 
     return history, history
 
+def generate_markdown(obj):
+    md_string = ""
+
+    if 'answer' in obj:
+        md_string += f"**Answer:**\n\n{obj['answer']}\n"
+
+    if 'sources' in obj:
+        sources_list = obj['sources'].strip().split('\n')
+        md_string += "**Sources:**\n\n"
+        for i, source in enumerate(sources_list):
+            md_string += f"{i + 1}. {source}\n"
+    
+    return md_string
 
 with gr.Blocks() as demo:
     gr.Markdown("<h3><center>BMW Chat Bot</center></h3>")