Update main.py

main.py

@@ -46,6 +46,12 @@ def create_gradio_interface():
 
     def chatbot_function(message, history, model, system_prompt, thinking_budget, current_stage):
         try:
+            # If this is the first message, get the initial greeting
+            if not history:
+                initial_greeting = innovative_chatbot.get_initial_greeting()
+                history.append((None, initial_greeting))
+                return history, "", ""
+
             chat_history, form_data = innovative_chatbot.process_stage_input(current_stage, message, model, system_prompt, thinking_budget)
             
             history.append((message, chat_history[-1][1]))
@@ -221,6 +227,10 @@ def create_gradio_interface():
             outputs=[chatbot, msg] + [form_fields[stage["name"]] for stage in STAGES] + [stages]
         )
 
+        # Add this new event handler to display the initial greeting when the interface loads
+        demo.load(lambda: ([[None, innovative_chatbot.get_initial_greeting()]], ""),
+                  outputs=[chatbot, msg])
+
     return demo
 
 def main():