flag to enable azure deployment

.gitignore

@@ -3,3 +3,4 @@ venv/*
 __pycache__/*
 .vscode/launch.json
 .vscode/settings.json
+.notes

app.py

@@ -23,17 +23,19 @@ unique_id = uuid4().hex[0:8]
 os.environ["LANGCHAIN_TRACING_V2"] = "true"
 os.environ["LANGCHAIN_PROJECT"] = f"LangSmith LCEL RAG - {unique_id}"
 
+is_azure = False if os.environ.get("AZURE_DEPLOYMENT") is None else True
+
 # Utility functions
-def save_file(file: AskFileResponse,file_ext:str) -> str:
+def save_file(file: AskFileResponse,file_ext:str,is_azure:bool) -> str:
     if file_ext == "application/pdf":
         file_ext = ".pdf"
     elif file_ext == "text/plain":
         file_ext = ".txt"
     else:
         raise ValueError(f"Unknown file type: {file_ext}")
-    
+    dir = "/tmp" if is_azure else None
     with tempfile.NamedTemporaryFile(
-        mode="wb", delete=False, suffix=file_ext
+        mode="wb", delete=False, suffix=file_ext,dir=dir
     ) as temp_file:
         temp_file_path = temp_file.name
         temp_file.write(file.content)
@@ -83,8 +85,8 @@ async def on_chat_start():
     ## Load file and split into chunks
     msg = cl.Message(content=f"Processing `{files[0].name}`...")
     await msg.send()
-    
-    current_file_path = save_file(files[0], files[0].type)
+
+    current_file_path = save_file(files[0], files[0].type,is_azure)
     loader_splitter = TextLoaderAndSplitterWrapper(ChunkingStrategy.RECURSIVE_CHARACTER_CHAR_SPLITTER, current_file_path)
     documents = loader_splitter.load_documents()