Spaces:
Running
Running
cha0smagick
commited on
Commit
•
1e2e8cd
1
Parent(s):
f6d566a
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,10 @@ def to_markdown(text):
|
|
11 |
# Function to generate content using Gemini API
|
12 |
def generate_gemini_content(prompt, model_name='gemini-pro', image=None):
|
13 |
model = genai.GenerativeModel(model_name)
|
|
|
|
|
|
|
|
|
14 |
if image:
|
15 |
response = model.generate_content([prompt, image])
|
16 |
else:
|
@@ -42,21 +46,23 @@ def main():
|
|
42 |
# Generate content on button click
|
43 |
if st.button("Generate Content"):
|
44 |
st.markdown("### Generated Content:")
|
45 |
-
if
|
46 |
-
|
47 |
-
|
48 |
-
|
|
|
|
|
49 |
else:
|
50 |
response = generate_gemini_content(prompt, model_name=model_name)
|
51 |
|
52 |
-
# Display the generated content in Markdown format
|
53 |
-
if response
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
|
61 |
if __name__ == "__main__":
|
62 |
main()
|
|
|
11 |
# Function to generate content using Gemini API
|
12 |
def generate_gemini_content(prompt, model_name='gemini-pro', image=None):
|
13 |
model = genai.GenerativeModel(model_name)
|
14 |
+
if model_name == 'gemini-pro-vision' and not image:
|
15 |
+
st.warning("Please add an image to use the gemini-pro-vision model.")
|
16 |
+
return None
|
17 |
+
|
18 |
if image:
|
19 |
response = model.generate_content([prompt, image])
|
20 |
else:
|
|
|
46 |
# Generate content on button click
|
47 |
if st.button("Generate Content"):
|
48 |
st.markdown("### Generated Content:")
|
49 |
+
if model_name == 'gemini-pro-vision':
|
50 |
+
if not image_file:
|
51 |
+
st.warning("Please provide an image for the gemini-pro-vision model.")
|
52 |
+
else:
|
53 |
+
image = Image.open(image_file)
|
54 |
+
response = generate_gemini_content(prompt, model_name=model_name, image=image)
|
55 |
else:
|
56 |
response = generate_gemini_content(prompt, model_name=model_name)
|
57 |
|
58 |
+
# Display the generated content in Markdown format if response is available
|
59 |
+
if response:
|
60 |
+
if response.candidates:
|
61 |
+
parts = response.candidates[0].content.parts
|
62 |
+
generated_text = parts[0].text if parts else "No content generated."
|
63 |
+
st.markdown(to_markdown(generated_text))
|
64 |
+
else:
|
65 |
+
st.warning("No candidates found in the response.")
|
66 |
|
67 |
if __name__ == "__main__":
|
68 |
main()
|