Add error handling to caption and focus generation functions in app_v4.py for improved robustness

app_v4.py

@@ -77,34 +77,46 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
     return image
 
 def combine_caption_focus(caption, focus):
-    if caption is None:
-        caption = ""
-    if focus is None:
-        focus = "highly detailed photo, raw photography."
-    return (str(caption) + "\n\n" + str(focus)).strip()
+    try:
+        if caption is None:
+            caption = ""
+        if focus is None:
+            focus = "highly detailed photo, raw photography."
+        return (str(caption) + "\n\n" + str(focus)).strip()
+    except Exception as e:
+        print(f"Error combining caption and focus: {e}")
+        return "highly detailed photo, raw photography."
 
 def generate_caption(control_image):
-    if control_image is None:
-        return "Waiting for control image..."
-    
-    # Generate a detailed caption
-    mcaption = model.caption(control_image, length="short")
-    detailed_caption = mcaption["caption"]
-    print(f"Detailed caption: {detailed_caption}")
-    
-    return detailed_caption
+    try:
+        if control_image is None:
+            return "Waiting for control image..."
+        
+        # Generate a detailed caption
+        mcaption = model.caption(control_image, length="short")
+        detailed_caption = mcaption["caption"]
+        print(f"Detailed caption: {detailed_caption}")
+        
+        return detailed_caption
+    except Exception as e:
+        print(f"Error generating caption: {e}")
+        return "A detailed photograph"
 
 def generate_focus(control_image, focus_list):
-    if control_image is None:
-        return None
-    if focus_list is None:
-        return ""
-    # Generate a detailed caption
-    focus_query = model.query(control_image, "Please provide a concise but illustrative description of the following area(s) of focus: " + focus_list)
-    focus_description = focus_query["answer"]
-    print(f"Areas of focus: {focus_description}") 
+    try:
+        if control_image is None:
+            return None
+        if focus_list is None:
+            return ""
+        # Generate a detailed caption
+        focus_query = model.query(control_image, "Please provide a concise but illustrative description of the following area(s) of focus: " + focus_list)
+        focus_description = focus_query["answer"]
+        print(f"Areas of focus: {focus_description}") 
 
-    return focus_description
+        return focus_description
+    except Exception as e:
+        print(f"Error generating focus: {e}")
+        return "highly detailed photo, raw photography."
 
 def process_image(control_image, user_prompt, system_prompt, scale, steps, 
                 controlnet_conditioning_scale, guidance_scale, seed,