Update app.py

app.py

@@ -32,6 +32,21 @@ SYSTEM_PROMPT = """You are an AI assistant specialized in analyzing images and v
 
 Always strive for accuracy in code extraction, as the user may need to directly use or analyze the extracted code."""
 
+@spaces.GPU
+def process_content(content, predefined_prompt, custom_prompt):
+    if content is None:
+        return "Please upload an image or video file of a code editor."
+
+    # Combine predefined and custom prompts
+    user_prompt = predefined_prompt if predefined_prompt else custom_prompt
+    full_prompt = f"{SYSTEM_PROMPT}\n\nUser request: {user_prompt}"
+
+    if content.name.lower().endswith(('.png', '.jpg', '.jpeg')):
+        return process_image(Image.open(content.name), full_prompt)
+    elif content.name.lower().endswith(('.mp4', '.avi', '.mov')):
+        return process_video(content, full_prompt)
+    else:
+        return "Unsupported file type. Please provide an image or video file of a code editor."
 
 @spaces.GPU
 def process_image(image, prompt):
@@ -127,20 +142,6 @@ def process_video(video, prompt, max_frames=16, frame_interval=30, max_resolutio
 
     return output_text[0]
 
-@spaces.GPU
-def process_content(content, prompt):
-    if content is None:
-        return "Please upload an image or video file of a code editor."
-
-    full_prompt = f"{SYSTEM_PROMPT}\n\nUser request: {prompt}"
-
-    if content.name.lower().endswith(('.png', '.jpg', '.jpeg')):
-        return process_image(Image.open(content.name), full_prompt)
-    elif content.name.lower().endswith(('.mp4', '.avi', '.mov')):
-        return process_video(content, full_prompt)
-    else:
-        return "Unsupported file type. Please provide an image or video file of a code editor."
-
 # Predefined prompts
 PREDEFINED_PROMPTS = [
     "Extract all visible code from this IDE screenshot.",