Update app.py

app.py

@@ -38,8 +38,11 @@ def generate_prompt(company_name, company_html, company_descp):
     return output
 
         
-def generate_image(comp_name, comp_desc, comp_url):
-    web_out=get_webpage_text(comp_url)
+def generate_image(comp_name, comp_desc, comp_url=""):
+    if comp_url:
+        web_out=get_webpage_text(comp_url)
+    else:
+        web_out="No Web Data returned"
     prompt_out=generate_prompt(comp_name,web_out, comp_desc)
     client = InferenceClient("black-forest-labs/FLUX.1-dev")
     response = client.text_to_image(prompt)