Upload app.py

app.py

@@ -1,5 +1,4 @@
 import requests
-import time
 import os
 
 import gradio as gr
@@ -23,8 +22,6 @@ EXAMPLE_QUERIES = [
 
 
 def make_request(query):
-    start_time = time.time()
-
     try:
         # Replace with your actual API endpoint
         if "rewrite: " not in query:
@@ -35,10 +32,7 @@ def make_request(query):
     except requests.exceptions.RequestException as e:
         result = f"Error: {str(e)}"
 
-    end_time = time.time()
-    processing_time = round(end_time - start_time, 2)
-
-    return result, f"Response Time: {processing_time} seconds"
+    return result
 
 
 # Create the Gradio interface
@@ -65,13 +59,12 @@ with gr.Blocks() as app:
 
     with gr.Row():
         response_output = gr.Textbox(label="Rewritten Text", lines=5, interactive=False)
-        time_label = gr.Label(label="Processing Time")
 
     # Add examples section
     gr.Examples(
         examples=EXAMPLE_QUERIES,
         inputs=query_input,
-        outputs=[response_output, time_label],
+        outputs=[response_output],
         fn=make_request,
         cache_examples=True,
         label="Example Queries",
@@ -80,18 +73,14 @@ with gr.Blocks() as app:
     # Clear button functionality
     clear_btn.click(
         lambda: ("", "", ""),  # Clear input  # Clear output  # Clear time label
-        outputs=[query_input, response_output, time_label],
+        outputs=[query_input, response_output],
     )
 
     # Submit button click event
-    submit_btn.click(
-        fn=make_request, inputs=[query_input], outputs=[response_output, time_label]
-    )
+    submit_btn.click(fn=make_request, inputs=[query_input], outputs=[response_output])
 
     # Add keyboard shortcut for submission
-    query_input.submit(
-        fn=make_request, inputs=[query_input], outputs=[response_output, time_label]
-    )
+    query_input.submit(fn=make_request, inputs=[query_input], outputs=[response_output])
 
 # Launch the app
 if __name__ == "__main__":