AiDeveloper1 commited on
Commit
3cf6da2
·
verified ·
1 Parent(s): bd0f463

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from flask import Flask,render_template,request,jsonify
2
- import openai
3
  from googletrans import Translator
4
  import asyncio
5
  from dotenv import load_dotenv
@@ -9,17 +9,14 @@ import os
9
  app = Flask(__name__)
10
 
11
 
12
- OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
13
- client = openai.Client(api_key=OPENAI_API_KEY)
14
 
15
  def get_response(prompt):
16
- stm = client.chat.completions.create(
17
- model="gpt-4o-mini",
18
- messages=[{"role": "user", "content": prompt}],
19
- stream=False,
20
- )
21
- res=stm.choices[0].message.content
22
- return res
23
 
24
 
25
  app = Flask(__name__)
 
1
  from flask import Flask,render_template,request,jsonify
2
+ import google.generativeai as genai
3
  from googletrans import Translator
4
  import asyncio
5
  from dotenv import load_dotenv
 
9
  app = Flask(__name__)
10
 
11
 
12
+ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
13
+ genai.configure(api_key=GEMINI_API_KEY)
14
 
15
  def get_response(prompt):
16
+ # Use Gemini model (e.g., gemini-1.5-flash)
17
+ model = genai.GenerativeModel('gemini-1.5-flash')
18
+ response = model.generate_content(prompt)
19
+ return response.text
 
 
 
20
 
21
 
22
  app = Flask(__name__)