Update app.py

app.py

@@ -3,7 +3,7 @@ import asyncio
 # Init with fake key
 if 'OPENAI_API_KEY' not in os.environ:
     os.environ['OPENAI_API_KEY'] = 'none'
-    os.environ["OPENAI_API_BASE"] = 'none'
+    
    
     os.environ["SERP_API_KEY"] =  'none'
     os.environ["SEMANTIC_SCHOLAR_API_KEY"] =  'none'
@@ -162,8 +162,19 @@ with st.sidebar:
         key='semantic_scholar_url',type='password',
         label_visibility="collapsed"
     ) 
-    os.environ['OPENAI_API_KEY'] = ss.get('api_key')
-    os.environ["OPENAI_API_BASE"] = ss.get('base_url')
+    user_api_key = ss.get('api_key')
+    user_base_url = ss.get('base_url')
+
+
+    if user_api_key:
+        os.environ['OPENAI_API_KEY'] = user_api_key
+
+    if user_base_url and user_base_url.strip() != "" and user_base_url != "none":
+        os.environ["OPENAI_API_BASE"] = user_base_url
+    else:
+      
+        if "OPENAI_API_BASE" in os.environ:
+            del os.environ["OPENAI_API_BASE"]
    
     os.environ["SERP_API_KEY"] =  ss.get('serp_api')
     os.environ["SEMANTIC_SCHOLAR_API_KEY"] =  ss.get('semantic_scholar_url')
@@ -257,4 +268,4 @@ if prompt := st.chat_input("Say something and/or attach files"):
                 # except:
                 #     st.markdown("Please input correct files or query ")
     else:
-          run_prompt(prompt)
+          run_prompt(prompt)