chore: refactor app.py for improved readability and performance

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 
 from autotab import AutoTab
 import json
+import time
+import pandas as pd
+
+
+def convert_seconds_to_time(seconds):
+    return time.strftime("%H:%M:%S", time.gmtime(seconds))
 
 
 def auto_tabulator_completion(
@@ -11,7 +17,9 @@ def auto_tabulator_completion(
     model_name,
     generation_config,
     save_every,
-):
+    api_key,
+    base_url,
+) -> tuple[str, str, str, pd.DataFrame]:
     output_file_name = "ouput.xlsx"
     autotab = AutoTab(
         in_file_path=in_file.name,
@@ -19,13 +27,16 @@ def auto_tabulator_completion(
         out_file_path=output_file_name,
         max_examples=max_examples,
         model_name=model_name,
-        api_key="sk-exhahhjfqyanmwewndukcqtrpegfdbwszkjucvcpajdufiah",
-        base_url="https://public-beta-api.siliconflow.cn/v1",
+        api_key=api_key,
+        base_url=base_url,
         generation_config=json.loads(generation_config),
         save_every=save_every,
     )
+    start = time.time()
     autotab.run()
-    return output_file_name, autotab.data[:15]
+    time_taken = time.strftime("%H:%M:%S", time.gmtime(time.time() - start))
+
+    return time_taken, output_file_name, autotab.query_example, autotab.data[:15]
 
 
 # Gradio interface
@@ -35,17 +46,23 @@ inputs = [
         value="You are a helpful assistant. Help me finish the task.",
         label="Instruction",
     ),
-    gr.Slider(value=5, minimum=1, maximum=100, label="Max Examples"),
+    gr.Slider(value=5, minimum=1, maximum=50, step=1, label="Max Examples"),
     gr.Textbox(value="Qwen/Qwen2-7B-Instruct", label="Model Name"),
     gr.Textbox(
         value='{"temperature": 0, "max_tokens": 128}',
         label="Generation Config in Dict",
     ),
-    gr.Slider(value=10, minimum=1, maximum=1000, label="Save Every N Steps"),
+    gr.Slider(value=100, minimum=1, maximum=1000, step=1, label="Save Every N Steps"),
+    gr.Textbox(
+        value="sk-exhahhjfqyanmwewndukcqtrpegfdbwszkjucvcpajdufiah", label="API Key"
+    ),
+    gr.Textbox(value="https://public-beta-api.siliconflow.cn/v1", label="Base URL"),
 ]
 
 outputs = [
+    gr.Textbox(label="Time Taken"),
     gr.File(label="Output Excel File"),
+    gr.Textbox(label="Query Example"),
     gr.Dataframe(label="First 15 rows."),
 ]