Spaces:
Running
Running
Update app.py
Browse files
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 |
+
|