datnguyentien204 commited on
Commit
78e2542
·
verified ·
1 Parent(s): 505cc89

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -2
app.py CHANGED
@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify, render_template
2
  from dotenv import load_dotenv
3
  import google.generativeai as genai
4
  import os
 
5
  load_dotenv()
6
  os.getenv("langchain_google_genai")
7
  os.environ['GOOGLE_API_KEY'] = 'AIzaSyAVFvEwx35owbCnkCpP5w2jbk6SXppFegs'
@@ -14,19 +15,39 @@ app = Flask(__name__)
14
  def home():
15
  return render_template('index.html')
16
 
 
 
 
 
17
  def question_answer(question):
18
  model = genai.GenerativeModel(model_name="gemini-pro")
19
  response = model.generate_content(question)
20
  return response.text
 
21
  @app.route('/send-message', methods=['POST'])
22
  def send_message():
23
  message = request.json.get('message')
24
  print(f"User message: {message}")
25
 
26
- # Simulate processing and generate a response
27
  response = question_answer(message)
28
  print(f"Ours: {response}")
29
  return jsonify({'response': response})
30
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  if __name__ == '__main__':
32
- app.run(host='0.0.0.0',port=8888,debug=True)
 
2
  from dotenv import load_dotenv
3
  import google.generativeai as genai
4
  import os
5
+
6
  load_dotenv()
7
  os.getenv("langchain_google_genai")
8
  os.environ['GOOGLE_API_KEY'] = 'AIzaSyAVFvEwx35owbCnkCpP5w2jbk6SXppFegs'
 
15
  def home():
16
  return render_template('index.html')
17
 
18
+ @app.route('/home2')
19
+ def home2():
20
+ return render_template('index_web2.html')
21
+
22
  def question_answer(question):
23
  model = genai.GenerativeModel(model_name="gemini-pro")
24
  response = model.generate_content(question)
25
  return response.text
26
+
27
  @app.route('/send-message', methods=['POST'])
28
  def send_message():
29
  message = request.json.get('message')
30
  print(f"User message: {message}")
31
 
 
32
  response = question_answer(message)
33
  print(f"Ours: {response}")
34
  return jsonify({'response': response})
35
 
36
+ @app.route('/login', methods=['POST'])
37
+ def login():
38
+ data = request.get_json()
39
+ email = data.get('email')
40
+ print(data)
41
+ password = data.get('password')
42
+
43
+ if email == 'abc@gmail' and password == 'abc@gmail':
44
+ message = f"Login successful with {email}"
45
+ print(message)
46
+ else:
47
+ message = "Invalid email or password"
48
+
49
+ return jsonify({'message': message})
50
+
51
+
52
  if __name__ == '__main__':
53
+ app.run(host='0.0.0.0', port=7860, debug=True)