Update app.py

app.py

@@ -28,6 +28,7 @@ def save_binary_file(file_name, data):
 def generate(text, api_key, model="gemini-2.0-flash-exp-image-generation"):
     logger.debug(f"Starting generate function with text: '{text}', model: '{model}'")
 
+    files = None  # Initialize 'files' to None
     try:
         # Initialize client
         effective_api_key = api_key.strip() if api_key and api_key.strip() != "" else os.environ.get("GEMINI_API_KEY")
@@ -37,9 +38,21 @@ def generate(text, api_key, model="gemini-2.0-flash-exp-image-generation"):
             logger.error("No API key provided or found in environment variable.")
             raise ValueError("API key is required.")
 
-        client = genai.Client(api_key=effective_api_key)
-        logger.debug("Gemini client initialized.")
-
+        try:
+            client = genai.Client(api_key=effective_api_key)
+            logger.debug("Gemini client initialized.")
+        except Exception as e:
+            logger.error(f"Error initializing Gemini client: {e}")
+            return None  # Return None if client initialization fails
+
+        try:
+            files = [
+                client.files.upload(file=text),  # Changed file=text
+            ]
+            logger.debug(f"File uploaded. URI: {files[0].uri}, MIME Type: {files[0].mime_type}")
+        except Exception as e:
+            logger.error(f"Error uploading file: {e}")
+            return None  # Return None if file upload fails
 
         contents = [
             types.Content(
@@ -86,18 +99,20 @@ def generate(text, api_key, model="gemini-2.0-flash-exp-image-generation"):
                     logger.info(f"File of mime type {inline_data.mime_type} saved to: {temp_path} and prompt input :{text}")
                 else:
                     logger.info(f"Received text: {chunk.text}")
-                    print(chunk.text) # Keep the print for immediate console output
+                    print(chunk.text)
 
                 # Log the raw chunk for deeper inspection
                 logger.debug(f"Raw chunk: {chunk}")
 
-        del files
+        # Check if files was assigned before deleting
+        if files:
+            del files
         logger.debug("Uploaded files deleted.")
         return temp_path
 
     except Exception as e:
-        logger.exception("An error occurred during generation:")  # This will log the full traceback
-        return None # Return None when error happens
+        logger.exception("An error occurred during generation:")
+        return None  # Return None when error happens
 
 
 def generate_image_from_prompt(prompt, gemini_api_key):
@@ -105,11 +120,11 @@ def generate_image_from_prompt(prompt, gemini_api_key):
     try:
 
         input_text = prompt
-        model = "gemini-2.0-flash-exp-image-generation"  #  Consider changing this to "gemini-pro-vision"
+        model = "gemini-2.0-flash-exp-image-generation"
 
         gemma_generated_image_path = generate(text=input_text, api_key=gemini_api_key, model=model)
 
-        if gemma_generated_image_path: # Check none or not
+        if gemma_generated_image_path:
             logger.debug(f"Image generated at path: {gemma_generated_image_path}")
             result_img = Image.open(gemma_generated_image_path)
             if result_img.mode == "RGBA":
@@ -117,11 +132,11 @@ def generate_image_from_prompt(prompt, gemini_api_key):
             return [result_img]
         else:
             logger.error("generate function returned None.")
-            return []  # Return empty when error
+            return []
 
     except Exception as e:
         logger.exception("Error occurred in generate_image_from_prompt")
-        return []  # Return empty when error
+        return []
 
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
@@ -165,8 +180,7 @@ with gr.Blocks() as demo:
     )
 
 try:
-    demo.launch(share=True)
+    demo.launch()
 except Exception as e:
     logger.error(f"Failed to launch Gradio app: {e}")
     print(f"Failed to launch Gradio app: {e}")
-