error handling

app.py

@@ -28,7 +28,7 @@ def scrape(URL, max_reviews=LIMIT):
             "basicInfo"
         ]["id"]
     except:
-        raise gr.Error("Invalid URL")
+        return "Invalid URL"
     else:
         all_reviews = []
         page = 1
@@ -80,14 +80,17 @@ def generate(URL, query):
         # Get reviews
         try:
             reviews = scrape(URL)
+            if (reviews is None) or (len(reviews) == 0):
+                return "No reviews found"
+            elif reviews == "Invalid URL":
+                return "Invalid URL"
             # Clean reviews
             cleaned_reviews = clean(reviews)
             # Load data
             loader = DataFrameLoader(cleaned_reviews, page_content_column="comment")
             documents = loader.load()
         except Exception as e:
-            print(e)
-            raise gr.Error("Error getting reviews: " + str(e))
+            return "Error getting reviews: " + str(e)
         else:
             # Split text
             text_splitter = RecursiveCharacterTextSplitter(
@@ -118,5 +121,6 @@ gr.Interface(
     outputs=gr.Textbox(label="Jawaban"),
     title="RingkasUlas",
     description="Bot percakapan yang bisa meringkas ulasan-ulasan produk di Tokopedia Indonesia (https://tokopedia.com/). Harap bersabar, bot ini dapat memakan waktu agak lama saat mengambil ulasan dari Tokopedia dan menyiapkan jawabannya.",
-    allow_flagging="never"
+    allow_flagging="never",
+    
 ).launch()