Update app.py

app.py

@@ -1,69 +1,31 @@
-import datetime
-import pytz
-import gradio as gr
 import os
 from huggingface_hub import InferenceClient
+import gradio as gr
 
-
-
-api_key = os.getenv("WingTokenAll")
-print(f"API Key Retrieved: {api_key is not None}")  # Debug line
-
-
-
-############################## Define tools Section ###############################
+# Function to test API connection
 def test_api_connection(query: str, api_key: str) -> str:
     """Tests the Hugging Face Inference API connection with the provided API key."""
     try:
-        # Initialize InferenceClient with the API key
         client = InferenceClient(model="distilbert-base-uncased-finetuned-sst-2-english", token=api_key)
-        # Send a test query to the model
         response = client.text_classification(query)
-        # Return the classification result or confirmation of success
         return f"API connection successful! Response: {response}"
     except Exception as e:
-        return f"Error: Failed to connect to Hugging Face API. Details: {str(e)}\n\nDebugging steps:\n1. Verify 'wingTokenRead' in Spaces Settings > Repository secrets matches the token from https://huggingface.co/settings/tokens.\n2. Ensure the token has read permissions and is not expired.\n3. Rebuild the Space to apply changes.\n4. If the issue persists, try a new token with read/write permissions."
-
-################################ Simple agent class #######################################
-class MyFirstAgent:
-
-    ###instance
-    def __init__(self):
-        self.tools = {
-            "test_api": test_api_connection
-        }
-
-    def run(self, query, api_key):
-
-        
-        query = query.lower().strip()
-
-        # Route to the API test tool
-        try:
-            if not query:
-                return "Error: Please provide a non-empty query."
-            return self.tools["test_api"](query, api_key)
-        except:
-            return "Error: Failed to process the query. Ensure the query and API key are valid."
-
-#################################### Create agent ##############################
-agent = MyFirstAgent()
+        return f"Error: {str(e)}"
 
-#@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ Define Gradio interface function #############################
+# Gradio interface function
 def run_agent(query):
-    # Retrieve API key from environment variable
     api_key = os.getenv("WingTokenAll")
     if not api_key:
-        return "Error: wingTokenRead not found in environment variables. Please set it in Spaces Settings > Repository secrets at https://huggingface.co/spaces."
-    return agent.run(query, api_key)
+        return "Error: wingTokenRead not found in environment variables."
+    return test_api_connection(query, api_key)
 
-#@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ Create and launch Gradio interface #########################################
+# Create and launch Gradio interface
 interface = gr.Interface(
     fn=run_agent,
     inputs=gr.Textbox(label="Enter a test query", placeholder="e.g., This is a test sentence"),
     outputs=gr.Textbox(label="API Test Response"),
-    title="My First Agent - Hugging Face API Test",
-    description="Enter a test query to verify the Hugging Face API connection using the wingTokenRead secret. Created on May 29, 2025, 11:42 PM HKT."
+    title="Hugging Face API Test",
+    description="Enter a test query to verify the Hugging Face API connection."
 )
 
 if __name__ == "__main__":