Docfile commited on
Commit
0615fa3
·
verified ·
1 Parent(s): 4a1d51a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +40 -20
app.py CHANGED
@@ -1,6 +1,9 @@
1
  from flask import Flask, request, render_template, jsonify
2
  import PIL.Image
3
- import google.generativeai as genai
 
 
 
4
  import os
5
  from tempfile import NamedTemporaryFile
6
  import tempfile
@@ -13,27 +16,31 @@ generation_config = {
13
  }
14
 
15
  safety_settings = [
16
- {
17
- "category": "HARM_CATEGORY_HARASSMENT",
18
- "threshold": "BLOCK_NONE"
19
- },
20
- {
21
- "category": "HARM_CATEGORY_HATE_SPEECH",
22
- "threshold": "BLOCK_NONE"
23
- },
24
- {
25
- "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
26
- "threshold": "BLOCK_NONE"
27
- },
28
- {
29
- "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
30
- "threshold": "BLOCK_NONE"
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