startup

app.py

@@ -2,6 +2,8 @@ from transformers import pipeline
 import gradio as gr
 import spaces
 
+# Hidden state for the counter
+counter_state = gr.State(0)
 
 pipe = pipeline("text-generation", model="markrodrigo/Llama-3.2-3B-Instruct-Spatial-SQL-1.1", device_map="auto")
 
@@ -66,7 +68,10 @@ custom_css = """
 
 
 @spaces.GPU
-def respond(user_message, chat_history):
+def respond(user_message, chat_history, counter):
+
+    # 1. Increment counter
+    counter += 1
 
     chat_history = chat_history or []
 
@@ -91,7 +96,7 @@ def respond(user_message, chat_history):
     bot_response = sequences[0]["generated_text"].strip()
     chat_history[-1]["content"] = bot_response
 
-    return chat_history, ""
+    return chat_history, "", counter
 
 
 
@@ -120,8 +125,9 @@ with gr.Blocks(title="Text to PostGIS Postgresql via Llama 3.2", css=custom_css)
         inputs=msg,
         label="Click an example → then click Submit"
     )
-    submit_btn.click(fn=respond, inputs=[msg, chatbot], outputs=[chatbot, msg])
-    msg.submit(fn=respond, inputs=[msg, chatbot], outputs=[chatbot, msg])
+    counter_display = gr.Textbox(label="Request Counter", value=0, interactive=False)
+    submit_btn.click(fn=respond, inputs=[msg, chatbot, counter_state], outputs=[chatbot, msg, counter_state, counter_display])
+    msg.submit(fn=respond, inputs=[msg, chatbot, counter_state], outputs=[chatbot, msg, counter_state, counter_display])
 
     clear_btn = gr.Button("Clear Chat", elem_id="clear-chat-btn")
     clear_btn.click(lambda: ([], ""), outputs=[chatbot, msg])