Updated app.py

app.py

@@ -423,18 +423,19 @@ def prompt_comb(input_file):
 
 # Define the function to call the model
 def call_model(model_name, input_file, prompt_name):
-    #index=int(prompt_name.split('prompt')[1]) - 1
-    prompt_combine=prompt_comb(input_file)
+    prompt_combine = prompt_comb(input_file)
+
+    if prompt_name == "Basic_v1":
+        index = 0
+    elif prompt_name == "Basic_v2":
+        index = 1  
+    elif prompt_name == "Generalize_v1":
+        index = 2 
+    elif prompt_name == "Generalize_v2":
+        index = 3          
+    
+    input_message = prompt_combine[index]
     
-    if prompt_name=="Basic_v1":
-        index=0
-    elif prompt_name=="Basic_v2":
-        index=1  
-    elif prompt_name=="Generalize_v1":
-        index=2 
-    elif prompt_name=="Generalize_v2":
-        index=3          
-    input_message=prompt_combine[index]
     # Call the selected model
     if model_name == "llama":
         output = call_model_llama(input_message)
@@ -444,6 +445,7 @@ def call_model(model_name, input_file, prompt_name):
         output = call_model_Gemini(input_message)
     else:
         output = call_model_Gemini1(input_message)
+    
     # Decode the output
     decoded_string = html.unescape(output)
     return decoded_string
@@ -456,14 +458,17 @@ def launch_interface():
     prompt_dropdown = gr.Dropdown(["Basic_v1", "Basic_v2", "Generalize_v1", "Generalize_v2"], label="Select Prompt")
 
     # Define the output component
-    output_text = gr.Textbox(label="Output")
-    # Create the Gradio interface
+    output_text = gr.Textbox(label="Output", interactive=False)
+
+    # Create the Gradio interface with dark theme
     interface = gr.Interface(
         fn=call_model,
         inputs=[model_dropdown, input_file, prompt_dropdown],
         outputs=output_text,
-        title="Part Identification Number(PINs)",
-        description="Choose a model **(llama-preferred)** and Upload a JSON file, select a prompt **(Generalize_v2-preffered)** , to generalize the PINs.",
+        title="Part Identification Number (PINs)",
+        description="Choose a model **(llama-preferred)** and Upload a JSON file, select a prompt **(Generalize_v2-preferred)**, to generalize the PINs.",
+        theme="dark",  # Set the theme to dark
+    
     )
 
     return interface