Update app.py

app.py

@@ -19,23 +19,21 @@ def generate(text, file_name, second_file_name, api_key, model="gemini-2.0-flash
     # Initialize client using provided api_key (or fallback to env variable)
     client = genai.Client(api_key=(api_key.strip() if api_key and api_key.strip() != ""
                                      else os.environ.get("GEMINI_API_KEY")))
-    
-    files = [ client.files.upload(file=file_name) ]
-    parts=[
-                types.Part.from_uri(
-                    file_uri=files[0].uri,
-                    mime_type=files[0].mime_type,
-                ),  
-               
+    parts = []
+    if file_name:
+        files = [ client.files.upload(file=file_name) ]
+        parts.append(types.Part.from_uri(
+            file_uri=files[0].uri,
+            mime_type=files[0].mime_type,
+        ))
                
-            ]
     if second_file_name:
         second_file =  client.files.upload(file=second_file_name)
         parts.append(types.Part.from_uri(
             file_uri=second_file.uri,
             mime_type=second_file.mime_type,
         ))
-        
+
     parts.append(types.Part.from_text(text=text))
     print('parts',parts)
     contents = [
@@ -86,11 +84,12 @@ def process_image_and_prompt(composite_pil, second_composite_pil, prompt, gemini
         # Save the composite image to a temporary file.
         # Get the file extension of the composite image
         # Get the format/extension from PIL image object
-       
-        with tempfile.NamedTemporaryFile(suffix=f".png", delete=False) as tmp:
-            composite_path = tmp.name
-            print('composite_path',composite_path)
-            composite_pil.save(composite_path)
+        composite_path = None
+        if composite_pil:
+            with tempfile.NamedTemporaryFile(suffix=f".png", delete=False) as tmp:
+                composite_path = tmp.name
+                print('composite_path',composite_path)
+                composite_pil.save(composite_path)
 
         second_file_name = None
         if second_composite_pil: