Update app.py

app.py

@@ -50,8 +50,8 @@ def read(filename):
     return data
 
 def process_text(text_input, unit):
-    endpoints = endpoints(api_key)
-    if text_input and endpoints == 'OPENAI':
+    endpoint = endpoints(api_key)
+    if text_input and endpoint == 'OPENAI':
         client = OpenAI(api_key=api_key)
         completion = client.chat.completions.create(
             model=model_name,
@@ -61,7 +61,7 @@ def process_text(text_input, unit):
             ]
         )
         return completion.choices[0].message.content
-    elif text_input and endpoints == "GOOGLE":
+    elif text_input and endpoint == "GOOGLE":
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel(model_name=model_name)
         prompt = f" You are a experienced {unit} doctor." + SYS_PROMPT + f"Could you solve {text_input}?"
@@ -76,8 +76,8 @@ def encode_image_to_base64(image_input):
     return img_str
 
 def process_image(image_input, unit):
-    endpoints = endpoints(api_key)
-    if image_input is not None and endpoints == 'OPENAI':
+    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:
         #    base64_image = base64.b64encode(f.read()).decode("utf-8")
@@ -99,7 +99,7 @@ def process_image(image_input, unit):
             max_tokens=1024,
         )
         return response.choices[0].message.content
-    elif image_input is not None and endpoints == "GOOGLE":
+    elif image_input is not None and endpoint == "GOOGLE":
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel(model_name=model_name)
         prompt = f" You are a experienced {unit} doctor." + SYS_PROMPT + "Help me understand what is in this picture and analysis."