update app.py file

app.py

@@ -58,15 +58,15 @@ class OMDbConnection:
 
         try:
             data = await self._get(params)
-        except Exception as e:
-            return f"ERROR: OMDb request failed: {type(e).__name__}: {e}"
+        except Exception:
+            return "ERROR: OMDb request failed. Please try again."
 
         # 2) If not found, fallback to search then fetch by imdbID
         if data.get("Response") != "True":
             try:
                 search = await self._get({"apikey": self.api_key, "s": clean_title, "type": "movie"})
-            except Exception as e:
-                return f"ERROR: OMDb search failed: {type(e).__name__}: {e}"
+            except Exception:
+                return "ERROR: OMDb request failed. Please try again."
 
             results = (search or {}).get("Search") or []
             if not results:
@@ -83,8 +83,8 @@ class OMDbConnection:
 
             try:
                 data = await self._get({"apikey": self.api_key, "i": imdb_id, "plot": "short"})
-            except Exception as e:
-                return f"ERROR: OMDb fetch-by-ID failed: {type(e).__name__}: {e}"
+            except Exception:
+                return "ERROR: OMDb fetch-by-ID failed. Please try again"
 
             if data.get("Response") != "True":
                 return f"ERROR: OMDb fetch-by-ID returned no result for imdbID={imdb_id}"
@@ -120,19 +120,22 @@ async def movie_system_prompt(ctx: RunContext[MovieData]) -> str:
     return await ctx.deps.omdb_conn.get_movie_info(ctx.deps.title)
 
 async def run_agent(user_query: str, movie_title: str):
-    deps = MovieData(title=movie_title, omdb_conn=OMDbConnection(OMDB_API_KEY))
-
-    # Preflight so we return clean UI errors (and avoid LLM call on missing data)
-    preflight = await deps.omdb_conn.get_movie_info(movie_title)
-    if preflight.startswith("ERROR:"):
-        return {"Error": preflight.replace("ERROR:", "").strip()}
-
-    result = await agent.run(user_query, deps=deps)
-    return {
-        "Why": result.output.why,
-        "Rating": int(result.output.rating),
-        "Recommend": bool(result.output.recommended),
-    }
+    try:
+        deps = MovieData(title=movie_title, omdb_conn=OMDbConnection(OMDB_API_KEY))
+
+        # Preflight so we return clean UI errors (and avoid LLM call on missing data)
+        preflight = await deps.omdb_conn.get_movie_info(movie_title)
+        if preflight.startswith("ERROR:"):
+            return {"Error": preflight.replace("ERROR:", "").strip()}
+
+        result = await agent.run(user_query, deps=deps)
+        return {
+            "Why": result.output.why,
+            "Rating": int(result.output.rating),
+            "Recommend": bool(result.output.recommended),
+        }
+    except Exception:
+        return {"Error": "Something went wrong. Please try again."}
 
 # ---- Gradio UI ----
 with gr.Blocks() as demo: