Update app.py

app.py

@@ -1,26 +1,32 @@
 import gradio as gr
 import requests
 
-def interact_with_server(prompt, ip_address, user_agent):
+def interact_with_server(prompt):
     server_url = "http://xzyorb.servemp3.com:80"
     response = requests.post(server_url, data={"prompt": prompt})
     response_text = response.text
     
     # Count the number of messages in the conversation
-    conversation_messages = response_text.split("\n")  # Assuming each message is separated by "<br>"
+    conversation_messages = response_text.split("<br>")  # Assuming each message is separated by "<br>"
     num_messages = len(conversation_messages)
     
-    # Print the information
+    return response_text, num_messages
+
+def print_session_data(data):
+    response_text, num_messages = data
+    session = gr.capture_session()
+    ip_address = session["ip"]
+    user_agent = session["user_agent"]
+    
     print("IP Address:", ip_address)
     print("User Agent:", user_agent)
     print("Number of Messages:", num_messages)
-    
-    return response_text
 
 iface = gr.Interface(
     fn=interact_with_server,
-    inputs=[gr.inputs.Textbox(), gr.inputs.Textbox(label="IP Address"), gr.inputs.Textbox(label="User Agent")],
-    outputs=gr.outputs.HTML()
+    inputs=gr.inputs.Textbox(),
+    outputs=[gr.outputs.HTML(), gr.outputs.Number()],  # Return both response text and number of messages
+    capture_session=True  # Automatically captures IP address and user agent
 )
 
-iface.launch()
+iface.launch(print_session_data)  # Pass the function to print session data to the launch method