victorgg commited on
Commit
1671f2b
·
verified ·
1 Parent(s): e645e5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -12
app.py CHANGED
@@ -45,15 +45,6 @@ def generate(text, api_key, model="gemini-2.0-flash-exp-image-generation"):
45
  logger.error(f"Error initializing Gemini client: {e}")
46
  return None # Return None if client initialization fails
47
 
48
- try:
49
- files = [
50
- client.files.upload(file=text), # Changed file=text
51
- ]
52
- logger.debug(f"File uploaded. URI: {files[0].uri}, MIME Type: {files[0].mime_type}")
53
- except Exception as e:
54
- logger.error(f"Error uploading file: {e}")
55
- return None # Return None if file upload fails
56
-
57
  contents = [
58
  types.Content(
59
  role="user",
@@ -104,9 +95,6 @@ def generate(text, api_key, model="gemini-2.0-flash-exp-image-generation"):
104
  # Log the raw chunk for deeper inspection
105
  logger.debug(f"Raw chunk: {chunk}")
106
 
107
- # Check if files was assigned before deleting
108
- if files:
109
- del files
110
  logger.debug("Uploaded files deleted.")
111
  return temp_path
112
 
@@ -138,6 +126,7 @@ def generate_image_from_prompt(prompt, gemini_api_key):
138
  logger.exception("Error occurred in generate_image_from_prompt")
139
  return []
140
 
 
141
  # --- Gradio Interface ---
142
  with gr.Blocks() as demo:
143
  gr.HTML(
@@ -184,3 +173,4 @@ try:
184
  except Exception as e:
185
  logger.error(f"Failed to launch Gradio app: {e}")
186
  print(f"Failed to launch Gradio app: {e}")
 
 
45
  logger.error(f"Error initializing Gemini client: {e}")
46
  return None # Return None if client initialization fails
47
 
 
 
 
 
 
 
 
 
 
48
  contents = [
49
  types.Content(
50
  role="user",
 
95
  # Log the raw chunk for deeper inspection
96
  logger.debug(f"Raw chunk: {chunk}")
97
 
 
 
 
98
  logger.debug("Uploaded files deleted.")
99
  return temp_path
100
 
 
126
  logger.exception("Error occurred in generate_image_from_prompt")
127
  return []
128
 
129
+
130
  # --- Gradio Interface ---
131
  with gr.Blocks() as demo:
132
  gr.HTML(
 
173
  except Exception as e:
174
  logger.error(f"Failed to launch Gradio app: {e}")
175
  print(f"Failed to launch Gradio app: {e}")
176
+