Smarter (doesn't crash when private API is not available)

README.md

@@ -1,7 +1,7 @@
 ---
-title: Property Embeddings Retrieval Public Endpoint
+title: TFG Web Demo Public API
 emoji: 🏙️
-colorFrom: gray
+colorFrom: purple
 colorTo: gray
 sdk: gradio
 sdk_version: 3.44.3

app.py

@@ -1,22 +1,34 @@
-import gradio
+import gradio, os
 from gradio_client import Client
-
 from dotenv import load_dotenv
+
 load_dotenv()
 
+client = None
 
 def redirect(query, collection_name):
-    import os
+    global client
+
     PRIVATE_SPACE_ID = os.getenv("PRIVATE_SPACE_ID")
     PRIVATE_API_KEY = os.getenv("PRIVATE_API_KEY")
 
-    client = Client(PRIVATE_SPACE_ID, hf_token=PRIVATE_API_KEY)
-    result = client.predict(
-        query,
-        collection_name,
-        api_name="/predict"
-    )
-    
+    try:
+        if client is None:
+            client = Client(PRIVATE_SPACE_ID, hf_token=PRIVATE_API_KEY)
+    except Exception as e:
+        print(f"Failed to connect to the client: {e}")
+        return "Failed to connect to the client. Please try again."
+
+    try:
+        result = client.predict(
+            query,
+            collection_name,
+            api_name="/predict"
+        )
+    except Exception as e:
+        print(f"Failed to get prediction: {e}")
+        return "Failed to get prediction. Please try again."
+
     return result
 
 
@@ -29,8 +41,7 @@ gradio_interface = gradio.Interface(
   ],
   title="REST API with Gradio and Huggingface Spaces",
   description="This is a REST API used for a project demo.",
-  article=""
+  article="© Fernando Ónega Rodrigo 2024"
 )
 
-gradio_interface.launch()
-
+gradio_interface.launch()