Update app.py

app.py

@@ -8,10 +8,9 @@ import base64
 
 # Set API key
 #base_url = os.environ.get("OPENAI_API_BASE")
-api_key = ""
     
 # Define the model to be used
-model_name = ""
+
 
 DESCRIPTION = '''
 <div>
@@ -37,20 +36,22 @@ footer {
 
 def endpoints(api_key):
     if api_key is not None:
-        if api_key.startswith(('sk-', 'sess-')):
+        if api_key.startswith('sk-'):
             model_name = "gpt-4o"
-            return 'OPENAI'
+            endpoint = 'OPENAI'
+            return model_name, endpoint
         else:
             model_name = "models/gemini-1.5-pro-latest"
-            return 'GOOGLE'
+            endpoint = 'GOOGLE'
+            return model_name, endpoint
 
 def read(filename):
     with open(filename) as f:
         data = f.read()
     return data
 
-def process_text(text_input, unit):
-    endpoint = endpoints(api_key)
+def process_text(api_key, text_input, unit):
+    model_name, endpoint = endpoints(api_key)
     if text_input and endpoint == 'OPENAI':
         client = OpenAI(api_key=api_key)
         completion = client.chat.completions.create(
@@ -75,8 +76,8 @@ def encode_image_to_base64(image_input):
     img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
     return img_str
 
-def process_image(image_input, unit):
-    endpoint = endpoints(api_key)
+def process_image(api_key, image_input, unit):
+    model_name, endpoint = endpoints(api_key)
     if image_input is not None and endpoint == 'OPENAI':
         client = OpenAI(api_key=api_key)
         #with open(image_input.name, "rb") as f: