Gosh this Leetmonkey

app.py

@@ -168,52 +168,29 @@ def api_random_problem(token: str) -> Dict[str, Any]:
     
     return {"problem": select_random_problem()}
 
-generate_interface = gr.Interface(
-    fn=api_generate_solution,
-    inputs=[
-        gr.Textbox(label="LeetCode Problem Instruction"),
-        gr.Textbox(label="JWT Token")
-    ],
-    outputs=gr.JSON(label="Generated Solution"),
-    title="Generate Solution API",
-    description="Provide a LeetCode problem instruction and a valid JWT token to generate a solution."
-)
-
-stream_interface = gr.Interface(
-    fn=api_stream_solution,
-    inputs=[
-        gr.Textbox(label="LeetCode Problem Instruction"),
-        gr.Textbox(label="JWT Token")
-    ],
-    outputs=gr.JSON(label="Streamed Solution"),
-    title="Stream Solution API",
-    description="Provide a LeetCode problem instruction and a valid JWT token to stream a solution."
-)
+def gradio_api(api_name: str, *args):
+    if api_name == "generate_solution":
+        return api_generate_solution(*args)
+    elif api_name == "stream_solution":
+        return api_stream_solution(*args)
+    elif api_name == "explain_solution":
+        return api_explain_solution(*args)
+    elif api_name == "random_problem":
+        return api_random_problem(*args)
+    else:
+        return {"error": "Invalid API name"}
 
-explain_interface = gr.Interface(
-    fn=api_explain_solution,
+iface = gr.Interface(
+    fn=gradio_api,
     inputs=[
-        gr.Textbox(label="Code to Explain"),
+        gr.Textbox(label="API Name"),
+        gr.Textbox(label="Problem Instruction"),
         gr.Textbox(label="JWT Token")
     ],
-    outputs=gr.JSON(label="Explanation"),
-    title="Explain Solution API",
-    description="Provide a code snippet and a valid JWT token to get an explanation."
-)
-
-random_problem_interface = gr.Interface(
-    fn=api_random_problem,
-    inputs=[gr.Textbox(label="JWT Token")],
-    outputs=gr.JSON(label="Random Problem"),
-    title="Random Problem API",
-    description="Provide a valid JWT token to get a random LeetCode problem."
-)
-
-# Combine interfaces
-demo = gr.TabbedInterface(
-    [generate_interface, stream_interface, explain_interface, random_problem_interface],
-    ["Generate Solution", "Stream Solution", "Explain Solution", "Random Problem"]
+    outputs=gr.JSON(label="API Response"),
+    title="LeetCode Problem Solver API",
+    description="Provide the API name, problem instruction (if required), and JWT token to use the desired functionality."
 )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    iface.launch(share=True)