Update app.py

app.py

@@ -13,7 +13,7 @@ api_key = ""
 endpoints = ''
     
 # Define the model to be used
-MODEL = os.environ.get("MODEL")
+model_name = ""
 
 DESCRIPTION = '''
 <div>
@@ -42,8 +42,10 @@ LICENSE = 'MODEL: ' + MODEL + ' LOADED'
 def endpoints(api_key):
     if api_key is not None:
         if api_key[:3] == "sk-":
+            model_name = "gpt-4o"
             return 'OPENAI'
         else:
+            model_name = "models/gemini-1.5-pro-latest"
             return 'GOOGLE'
     return
 
@@ -60,7 +62,7 @@ def process_text(text_input, unit):
     if text_input and endpoints == 'OPENAI':
         client = OpenAI(api_key=api_key)
         completion = client.chat.completions.create(
-            model=MODEL,
+            model=model_name,
             messages=[
                 {"role": "system", "content": f" You are a experienced {unit} doctor." + SYS_PROMPT},
                 {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
@@ -69,7 +71,7 @@ def process_text(text_input, unit):
         return completion.choices[0].message.content
     elif text_input and endpoints == "GOOGLE":
         genai.configure(api_key=api_key)
-        model = genai.GenerativeModel(MODEL),
+        model = genai.GenerativeModel(model_name=model_name),
         response = model.generate_content(f" You are a experienced {unit} doctor." + SYS_PROMPT + f"Could you solve {text_input}?")
         return response.text
     return ""
@@ -87,7 +89,7 @@ def process_image(image_input, unit):
         #    base64_image = base64.b64encode(f.read()).decode("utf-8")
         base64_image = encode_image_to_base64(image_input)
         response = client.chat.completions.create(
-            model=MODEL,
+            model=model_name,
             messages=[
                 {"role": "system", "content": f" You are a experienced {unit} doctor." + SYS_PROMPT},
                 {"role": "user", "content": [
@@ -103,9 +105,9 @@ 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 endpoints == "GOOGLE":
         genai.configure(api_key=api_key)
-        model = genai.GenerativeModel(MODEL),
+        model = genai.GenerativeModel(model_name=model_name),
         response = model.generate_content([f" You are a experienced {unit} doctor." + SYS_PROMPT + "Help me understand what is in this picture and analysis.", image_input])
         return response.text
 
@@ -126,7 +128,7 @@ with gr.Blocks(theme='shivi/calm_seafoam', css=css, title="Medster - Medical Dia
     with gr.Row():
         output_box = gr.Markdown(label="Diagnosis")  # Create an output textbox
     with gr.Row():
-        api_key = gr.Textbox(label="API Key") # Input API key
+        api_key = gr.Textbox(label="API Key", type='password') # Input API key
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload Image")  # Create an image upload button
         text_input = gr.Textbox(label="Submit")  # Create a text input box