Update app.py

app.py

@@ -99,10 +99,11 @@ with gr.Blocks() as demo:
     def user(user_message, history):
         # Get response from QA chain
         docs = docsearch.similarity_search(user_message)
-        output = chain({"input_documents": docs, "human_input": user_message}, return_only_outputs=True)
+        output = chain({"input_documents": docs, "human_input": user_message}, return_only_outputs=False)
 
         model_info = get_image_url(user_message)
         output_text = output['output_text']
+        output_text = re.sub(r'^\W+', '', output_text)
         
         # Construct the HTML for displaying images
         images_html = ""
@@ -112,8 +113,13 @@ with gr.Blocks() as demo:
                 images_html += f"Toyota Tundra {model_name.capitalize()} 2023: {image_html}"
         
         # Adding the source link
-        source_link = "" #"Source: [Toyota Tundra 2023](https://www.toyota.com/tundra/2023/)"
-        output_text_with_images = f"{output_text}<br>{images_html}{source_link}"
+        if docs[0].page_content == 'nan' or docs[1].page_content == 'nan' or docs[0].page_content == '♪':
+            video_link = ""
+        else:
+            video_link = f"Source: {output['input_documents'][0].metadata['video_link']}"
+            
+        
+        output_text_with_images = f"{output_text}<br>{images_html}{video_link}"
         
         history.append((user_message, output_text_with_images))
         return gr.update(value=""), history