Update app.py

app.py

@@ -3,7 +3,6 @@ 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
@@ -14,7 +13,7 @@ from haystack_integrations.components.retrievers.mongodb_atlas import MongoDBAtl
 from haystack.components.embedders import OpenAITextEmbedder, OpenAIDocumentEmbedder
 from haystack.document_stores.types import DuplicatePolicy
 from haystack.components.writers import DocumentWriter
-
+import requests
 import json
 
 
@@ -88,9 +87,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):
@@ -115,15 +114,7 @@ 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
-           
-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}"
+headers = ["Plot"]
 
 def update_movie_data(new_data):
     """
@@ -141,20 +132,31 @@ def update_movie_data(new_data):
     indexing_pipe.run({"doc_embedder": {"documents": new_documents}})
 
     return new_data
+
+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}"
+        
 # Setup Gradio interface
 with gr.Blocks() as demo:
-        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("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("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()