Update app.py

app.py

@@ -3,6 +3,7 @@ import traceback
 import gradio as gr
 import os
 import asyncio
+import requests
 from haystack_integrations.document_stores.mongodb_atlas import MongoDBAtlasDocumentStore
 from haystack import Pipeline, Document
 from haystack.components.generators import OpenAIGenerator
@@ -87,9 +88,9 @@ try:
     rag_pipeline.connect("retriever", "prompt_builder.documents")
     rag_pipeline.connect("prompt_builder", "llm")
     
-    # Exception handling to catch and display errors during the pipeline execution.
-    except Exception as e:
-        print("An error occurred: \n" + error_message)
+# Exception handling to catch and display errors during the pipeline execution.
+except Exception as e:
+    print("An error occurred: \n" + error_message)
     
 
 def get_movies(message, history):
@@ -114,7 +115,15 @@ def get_movies(message, history):
         
 # Convert documents to a format suitable for Gradio Dataframe
 data_for_dataframe = [[doc.content] for doc in documents]
-headers = ["Plot"]
+headers = ["Plot
+           
+def fetch_url_data(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # Raises an HTTPError if the HTTP request returned an unsuccessful status code
+        return response.text
+    except requests.RequestException as e:
+        return f"Error: {e}"
 
 def update_movie_data(new_data):
     """
@@ -134,15 +143,18 @@ def update_movie_data(new_data):
     return new_data
 # Setup Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## Movie Plot Viewer")
-    movie_table = gr.Dataframe(value=data_for_dataframe, headers=headers, interactive=False)
-    submit_button = gr.Button("Update Data")
-    ## value=[(None, "Hi, I'm a MongoDB and Heystack based question and answer bot 🤖, I can help you answer on the knowledge base above…")]
-    gr.ChatInterface(get_movies,examples=["What characters are from Rome?", "Combine 3 plots of your choice", "List all characters"], title="Atlas Vector Search Chat",description="This small chat uses a similarity search to find relevant plots as listed above, it uses MongoDB Atlas and Haystack integaration: https://haystack.deepset.ai/integrations/mongodb",submit_btn="Search").queue()
-
-    
-
-    submit_button.click(fn=update_movie_data, inputs=[movie_table], outputs=[movie_table])
-
+        with gr.Tab("Chat"):
+            gr.Markdown("## Movie Plot Viewer")
+            movie_table = gr.Dataframe(value=data_for_dataframe, headers=headers, interactive=False)
+            submit_button = gr.Button("Update Data")
+            ## value=[(None, "Hi, I'm a MongoDB and Heystack based question and answer bot 🤖, I can help you answer on the knowledge base above…")]
+            gr.ChatInterface(get_movies,examples=["What characters are from Rome?", "Combine 3 plots of your choice", "List all characters"], title="Atlas Vector Search Chat",description="This small chat uses a similarity search to find relevant plots as listed above, it uses MongoDB Atlas and Haystack integaration: https://haystack.deepset.ai/integrations/mongodb",submit_btn="Search").queue()
+        
+            
+        
+            submit_button.click(fn=update_movie_data, inputs=[movie_table], outputs=[movie_table])
+    with gr.Tab("Code"):
+        gr.Code(label="Code", language="python", value=fetch_url_data('https://huggingface.co/spaces/MongoDB/Haystack-MongoDB-Integration-Chat/raw/main/app.py'))
+        
 if __name__ == "__main__":
     demo.launch()