Update app.py

app.py

@@ -43,6 +43,7 @@ body::before {
 available_models = [
     "hf://ShebMichel/geobot_teacher-v0",
     "hf://ShebMichel/geobot_teacher-v1"]
+first_model = available_models[0]  # Set the first item as a global variable
 
 # geomodel_llm = keras_nlp.models.CausalLM.from_preset("hf://ShebMichel/geobot_teacher-v0")
 # Function to load the selected model
@@ -53,7 +54,7 @@ def load_model(model_name):
 
 
 
-def model_launch(input,selected_output):
+def model_launch(input,first_model):
     return f"Model: {selected_output}\nResponse to: {input}"
     # geomodel_llm=load_model(selected_model)
     # template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
@@ -163,6 +164,12 @@ def submit_question(question_content):
 def display_selected_model(selected_model):
     return f"Selected Model: {selected_model}"
 
+# Function to update the global `first_model` whenever the dropdown changes
+def update_first_model(selected_model):
+    global first_model
+    first_model = selected_model
+    return f"Selected Model: {first_model}"
+
 def clear_output():
     return "", ""
 
@@ -202,17 +209,33 @@ with gr.Blocks() as iface:
             model_dropdown = gr.Dropdown(
                 choices=available_models,
                 label="Select Model",
-                value=None
+                value=first_model  # Set initial selection to the first model
                # visible=False  # Initially hidden
             )
             # Create a Textbox to display the selected model
-            selected_output = gr.Textbox(label="Selected Model", interactive=False)
+            #selected_output = gr.Textbox(label="Selected Model", interactive=False)
             # Set up an event to update the Textbox when a model is selected
-        model_dropdown.change(
-            fn=display_selected_model,
-            inputs=model_dropdown,
-            outputs=selected_output
-        )
+        # model_dropdown.change(
+        #     fn=display_selected_model,
+        #     inputs=model_dropdown,
+        #     outputs=selected_output
+        # )
+
+            #         # Create the dropdown for model selection
+            # model_dropdown = gr.Dropdown(
+            #     choices=available_models,
+            #     label="Select Model",
+            #     value=first_model  # Set initial selection to the first model
+            # )
+            # Textbox to display the current selected model
+            selected_output = gr.Textbox(label="Current Selected Model", value=f"Selected Model: {first_model}")
+
+            # Update the global `first_model` when dropdown changes and display it
+            model_dropdown.change(
+                fn=update_first_model,
+                inputs=model_dropdown,
+                outputs=selected_output
+            )
         #select_button = gr.Button("Select Model", variant="primary")
         
     # Editable question text box
@@ -244,7 +267,7 @@ with gr.Blocks() as iface:
 
     submit_button.click(
         fn=model_launch,#submit_question, ## model_launch
-        inputs=[question_edit, selected_output],  # Include the dropdown in inputs
+        inputs=[question_edit, first_model],  # Include the dropdown in inputs
         outputs=[model_response] 
     )