Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,9 @@
|
|
1 |
from flask import Flask, request, render_template, jsonify
|
2 |
import PIL.Image
|
3 |
-
|
|
|
|
|
|
|
4 |
import os
|
5 |
from tempfile import NamedTemporaryFile
|
6 |
import tempfile
|
@@ -13,27 +16,31 @@ generation_config = {
|
|
13 |
}
|
14 |
|
15 |
safety_settings = [
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
]
|
33 |
|
34 |
GOOGLE_API_KEY = os.environ.get("TOKEN")
|
35 |
|
36 |
-
genai.configure(api_key=GOOGLE_API_KEY)
|
|
|
|
|
|
|
|
|
37 |
|
38 |
@app.route('/', methods=['GET'])
|
39 |
def svt():
|
@@ -145,8 +152,21 @@ def svt_submit():
|
|
145 |
temp_files.append(temp_file.name)
|
146 |
content.append(PIL.Image.open(temp_file.name))
|
147 |
|
148 |
-
model = genai.GenerativeModel(model_name="models/gemini-2.5-flash-preview-04-17", safety_settings=safety_settings)
|
149 |
-
response = model.generate_content(content, request_options={"timeout": 600})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
150 |
|
151 |
return jsonify({"response": response.text})
|
152 |
|
|
|
1 |
from flask import Flask, request, render_template, jsonify
|
2 |
import PIL.Image
|
3 |
+
from google import genai
|
4 |
+
from google.genai import types
|
5 |
+
|
6 |
+
|
7 |
import os
|
8 |
from tempfile import NamedTemporaryFile
|
9 |
import tempfile
|
|
|
16 |
}
|
17 |
|
18 |
safety_settings = [
|
19 |
+
types.SafetySetting(
|
20 |
+
category="HARM_CATEGORY_HARASSMENT",
|
21 |
+
threshold="BLOCK_NONE",
|
22 |
+
),
|
23 |
+
types.SafetySetting(
|
24 |
+
category="HARM_CATEGORY_HATE_SPEECH",
|
25 |
+
threshold="BLOCK_NONE",
|
26 |
+
),
|
27 |
+
types.SafetySetting(
|
28 |
+
category="HARM_CATEGORY_SEXUALLY_EXPLICIT",
|
29 |
+
threshold="BLOCK_NONE",
|
30 |
+
),
|
31 |
+
types.SafetySetting(
|
32 |
+
category="HARM_CATEGORY_DANGEROUS_CONTENT",
|
33 |
+
threshold="BLOCK_NONE",
|
34 |
+
),
|
35 |
]
|
36 |
|
37 |
GOOGLE_API_KEY = os.environ.get("TOKEN")
|
38 |
|
39 |
+
#genai.configure(api_key=GOOGLE_API_KEY)
|
40 |
+
|
41 |
+
gen = GOOGLE_API_KEY
|
42 |
+
client = genai.Client(api_key=gen)
|
43 |
+
|
44 |
|
45 |
@app.route('/', methods=['GET'])
|
46 |
def svt():
|
|
|
152 |
temp_files.append(temp_file.name)
|
153 |
content.append(PIL.Image.open(temp_file.name))
|
154 |
|
155 |
+
#model = genai.GenerativeModel(model_name="models/gemini-2.5-flash-preview-04-17", safety_settings=safety_settings)
|
156 |
+
#response = model.generate_content(content, request_options={"timeout": 600})
|
157 |
+
|
158 |
+
response = client.models.generate_content(
|
159 |
+
model="gemini-2.5-flash-preview-04-17",
|
160 |
+
contents=[content],
|
161 |
+
config=types.GenerateContentConfig(
|
162 |
+
safety_settings=safety_settings,
|
163 |
+
|
164 |
+
)
|
165 |
+
),
|
166 |
+
)
|
167 |
+
|
168 |
+
answer = response.candidates[0].content.parts[0].text
|
169 |
+
return answer.strip()
|
170 |
|
171 |
return jsonify({"response": response.text})
|
172 |
|