Athspi commited on
Commit
3ebc508
·
verified ·
1 Parent(s): 04e9417

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -0
app.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app.py - Flask Backend
2
+ from flask import Flask, request, jsonify
3
+ import google.generativeai as genai
4
+ from dotenv import load_dotenv
5
+ import os
6
+ from flask_cors import CORS
7
+
8
+ # Load environment variables
9
+ load_dotenv()
10
+
11
+ # Initialize Flask app
12
+ app = Flask(__name__)
13
+ CORS(app) # Enable CORS for all routes
14
+
15
+ # Configure Gemini
16
+ genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
17
+ model = genai.GenerativeModel('gemini-pro')
18
+
19
+ @app.route('/chat', methods=['POST'])
20
+ def chat():
21
+ try:
22
+ data = request.json
23
+ user_message = data.get('message')
24
+
25
+ if not user_message:
26
+ return jsonify({"error": "No message provided"}), 400
27
+
28
+ # Generate response using Gemini
29
+ response = model.generate_content(user_message)
30
+
31
+ return jsonify({
32
+ "response": response.text
33
+ })
34
+
35
+ except Exception as e:
36
+ return jsonify({"error": str(e)}), 500
37
+
38
+ @app.route('/')
39
+ def serve_index():
40
+ return app.send_static_file('index.html')
41
+
42
+ if __name__ == '__main__':
43
+ app.run(host='0.0.0.0', port=5000)