Update app.py

app.py

@@ -17,7 +17,6 @@ from llama_index.core import SimpleDirectoryReader
 from llama_index.core.indices.multi_modal.base import MultiModalVectorStoreIndex
 from llama_index.multi_modal_llms.openai import OpenAIMultiModal
 
-os.environ["OPENAI_API_KEY"] = "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"
 embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
 chat_llm = ChatOpenAI(temperature = 0.5, model = 'gpt-4-turbo')
 
@@ -47,9 +46,10 @@ qa_prompt = ChatPromptTemplate.from_messages(
 )
 question_answer_chain = create_stuff_documents_chain(chat_llm, qa_prompt)
 
-# pg_connection = "postgresql+psycopg://postgres:3434@localhost:5433/mmrag"
-pg_connection = "postgresql+psycopg://postgres:3434@43.201.34.191:5432/postgres"
-qd_client = qdrant_client.QdrantClient(path="qdrant_db") 
+pg_password = os.getenv("PG_PASSWORD")
+aws_ec2_ip = os.getenv("AWS_EC2_IP")
+pg_connection = f"postgresql+psycopg://postgres:{pg_password}@{aws_ec2_ip}:5432/postgres"
+qd_client = qdrant_client.QdrantClient(path="qdrant_db")
 image_store = QdrantVectorStore(client=qd_client, collection_name="image_collection")
 storage_context = StorageContext.from_defaults(image_store=image_store)
 openai_mm_llm = OpenAIMultiModal(model="gpt-4o", max_new_tokens=1500)
@@ -96,8 +96,8 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
             doc_label = gr.Dropdown(["LLaVA", "Interior"], label="Select a document:")
             chatbot = gr.ChatInterface(fn=response, additional_inputs=[doc_label], fill_height=True)
         with gr.Column(scale=1):
-            sample_1 = "https://i.ytimg.com/vi/bLj_mR4Fnls/maxresdefault.jpg"
-            sample_2 = "https://i.ytimg.com/vi/bOJdHU99OO8/maxresdefault.jpg"
+            sample_1 = "https://i.pinimg.com/originals/e3/44/d7/e344d7631cd515edd36cc6930deaedec.jpg"
+            sample_2 = "https://live.staticflickr.com/5307/5765340890_e386f42a99_b.jpg"
             sample_3 = "https://blog.kakaocdn.net/dn/nqcUB/btrzYjTgjWl/jFFlIBrdkoKv4jbSyZbiEk/img.jpg"
             gallery = gr.Gallery(label="Retrieved images", 
                                  show_label=True, preview=True,