Update app.py

app.py

@@ -178,6 +178,101 @@ def respond(message, history, system_message):
 # Update the system prompt to be more relevant to maritime legal assistance
 system_prompt = "You are a maritime legal assistant with expertise in maritime law. Provide detailed legal advice and information based on maritime legal principles and regulations."
 
+# Use gr.Blocks and gr.Chatbot for Gradio 3.x
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    system_message_input = gr.Textbox(value=system_prompt, label="System message")
+    message_input = gr.Textbox(label="Message")
+    
+    def process_message(message, history, system_message):
+        response_gen = respond(message, history, system_message)
+        return list(response_gen), history
+
+    message_input.submit(process_message, [message_input, chatbot, system_message_input], [chatbot, chatbot])
+
+if __name__ == "__main__":
+    demo.launch()
+
+    assistant_content = ""
+    assistant_native_message_list = []
+
+    while True:
+        response_message = (
+            get_model_response(history, inner_history, message, system_message)
+            .choices[0]
+            .message
+        )
+
+        if not response_message.tool_calls and response_message.content is not None:
+            break
+
+        if response_message.tool_calls is not None:
+            assistant_native_message_list.append(response_message)
+            inner_history.append(response_message)
+
+            assistant_content += (
+                "```json\n"
+                + json.dumps(
+                    [
+                        tool_call.model_dump()
+                        for tool_call in response_message.tool_calls
+                    ],
+                    indent=2,
+                )
+                + "\n```\n"
+            )
+            assistant_message = {
+                "role": "assistant",
+                "content": assistant_content,
+                "metadata": {"native_messages": assistant_native_message_list},
+            }
+
+            yield assistant_message
+
+            for tool_call in response_message.tool_calls:
+                function_response = call_function(tool_call, available_functions)
+                assistant_content += (
+                    "```json\n"
+                    + json.dumps(
+                        {
+                            "name": tool_call.function.name,
+                            "arguments": json.loads(tool_call.function.arguments),
+                            "response": json.loads(function_response["content"]),
+                        },
+                        indent=2,
+                    )
+                    + "\n```\n"
+                )
+                native_tool_message = {
+                    "tool_call_id": tool_call.id,
+                    "role": "tool",
+                    "content": function_response["content"],
+                }
+                assistant_native_message_list.append(
+                    native_tool_message
+                )
+                tool_message = {
+                    "role": "assistant",
+                    "content": assistant_content,
+                    "metadata": {"native_messages": assistant_native_message_list},
+                }
+                yield tool_message
+                inner_history.append(native_tool_message)
+
+    assistant_content += response_message.content
+    assistant_native_message_list.append(response_message)
+
+    final_message = {
+        "role": "assistant",
+        "content": assistant_content,
+        "metadata": {"native_messages": assistant_native_message_list},
+    }
+    
+    yield final_message
+
+# Update the system prompt to be more relevant to maritime legal assistance
+system_prompt = "You are a maritime legal assistant with expertise in maritime law. Provide detailed legal advice and information based on maritime legal principles and regulations."
+
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[