Neurolingua commited on
Commit
c2ed2b0
·
verified ·
1 Parent(s): cb78502

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +172 -171
app.py CHANGED
@@ -1,171 +1,172 @@
1
- from flask import Flask, render_template, request, jsonify, send_from_directory
2
- import pymysql
3
- import matplotlib.pyplot as plt
4
- import io
5
- import base64
6
- from gtts import gTTS
7
- import os
8
- import random
9
- from ai71 import AI71
10
-
11
- app = Flask(__name__)
12
- db = pymysql.connect(host="localhost", user="root", password="1234", database="tea_farm")
13
- cursor = db.cursor()
14
-
15
- UPLOAD_FOLDER = 'uploads'
16
- OUTPUT_DIRECTORY = 'audio'
17
- if not os.path.exists(UPLOAD_FOLDER):
18
- os.makedirs(UPLOAD_FOLDER)
19
- if not os.path.exists(OUTPUT_DIRECTORY):
20
- os.makedirs(OUTPUT_DIRECTORY)
21
-
22
- # Text for pest and disease
23
- pest_text = '''tea mosquito bugs,to get rid,spray with neem oil and garlic,use sticky traps, and trap crops like castor . Expect results in 2 to 4 weeks.চা মশার বাগ, পরিত্রাণ পেতে, নিম তেল এবং রসুন দিয়ে স্প্রে করুন, আঠালো ফাঁদ ব্যবহার করুন এবং ক্যাস্টরের মতো ফসল ফাঁদ করুন। 2-4 সপ্তাহের মধ্যে ফলাফল আশা করুন।'''
24
- disease_text = '''Detected Disease: Brown Blight is a fungal disease To treat it spray a mix of baking soda, water and apply organic mulch to keep soil healthy.ব্রাউন ব্লাইট একটি ছত্রাকজনিত রোগের চিকিৎসার জন্য বেকিং সোডা, পানির মিশ্রণ স্প্রে করুন এবং মাটি সুস্থ রাখতে জৈব মালচ প্রয়োগ করুন'''
25
-
26
- @app.route('/home')
27
- def home():
28
- return render_template('index.html')
29
-
30
- @app.route('/notification')
31
- def notification():
32
- return render_template('notification.html')
33
-
34
- @app.route('/', methods=['GET', 'POST'])
35
- def upload_file():
36
- pest_filename = None
37
- disease_filename = None
38
- pest_audio_file = "pest.mp3"
39
- disease_audio_file = "disease.mp3"
40
-
41
- if request.method == 'POST':
42
- if 'pest_file' in request.files:
43
- pest_file = request.files['pest_file']
44
- if pest_file.filename != '':
45
- pest_filename = pest_file.filename
46
- pest_file.save(os.path.join(UPLOAD_FOLDER, pest_filename))
47
- tts_pest = gTTS(text=pest_text, lang='en', slow=False)
48
- tts_pest.save(os.path.join(OUTPUT_DIRECTORY, pest_audio_file))
49
-
50
- if 'disease_file' in request.files:
51
- disease_file = request.files['disease_file']
52
- if disease_file.filename != '':
53
- disease_filename = disease_file.filename
54
- disease_file.save(os.path.join(UPLOAD_FOLDER, disease_filename))
55
- tts_disease = gTTS(text=disease_text, lang='en', slow=False)
56
- tts_disease.save(os.path.join(OUTPUT_DIRECTORY, disease_audio_file))
57
-
58
- return render_template('index.html', pest_filename=pest_filename, disease_filename=disease_filename,
59
- pest_audio_file=pest_audio_file, disease_audio_file=disease_audio_file)
60
-
61
- AI71_API_KEY = "api71-api-20725a9d-46d6-4baf-9e26-abfca35ab242"
62
- @app.route('/chat', methods=['POST'])
63
- def chat():
64
- message = request.json['message']
65
- ai71 = AI71(AI71_API_KEY)
66
- response = ""
67
- for chunk in ai71.chat.completions.create(
68
- model="tiiuae/falcon-180b-chat",
69
- messages=[
70
- {"role": "system", "content": "You are a helpful assistant for a farming system.Guide the user properly to increase the yield."},
71
- {"role": "user", "content": message},
72
- ],
73
- stream=True,
74
- ):
75
- if chunk.choices[0].delta.content:
76
- response += chunk.choices[0].delta.content
77
- return jsonify({'response': response.replace('\n','<br>')})
78
- @app.route('/uploads/<filename>')
79
- def uploaded_file(filename):
80
- return send_from_directory(UPLOAD_FOLDER, filename)
81
-
82
- @app.route('/audio/<filename>')
83
- def audio_file(filename):
84
- return send_from_directory(OUTPUT_DIRECTORY, filename)
85
-
86
- @app.route('/sensors')
87
- def sensors():
88
- # Fetch a random row from the database
89
- cursor.execute("SELECT * FROM sensor_data ORDER BY RAND() LIMIT 1")
90
- sensor_data = cursor.fetchone()
91
-
92
- if sensor_data:
93
- sensor_dict = {
94
- 'id': sensor_data[0],
95
- 'temperature': sensor_data[1],
96
- 'humidity': sensor_data[2],
97
- 'irValue': sensor_data[3],
98
- 'distance': sensor_data[4],
99
- 'timestamp': sensor_data[5]
100
- }
101
- else:
102
- sensor_dict = {
103
- 'temperature': None,
104
- 'humidity': None,
105
- 'irValue': None,
106
- 'distance': None,
107
- 'timestamp': None
108
- }
109
-
110
- return render_template('sensors.html', sensor_data=sensor_dict)
111
-
112
- @app.route('/zones')
113
- def zones():
114
- return render_template('zones.html')
115
-
116
- @app.route('/activity-log')
117
- def activity_log():
118
-
119
- # Fetch data for IR value distribution plot
120
- cursor.execute("SELECT ir_value FROM sensor_data")
121
- ir_values = cursor.fetchall()
122
- count_0 = ir_values.count((0,))
123
- count_1 = ir_values.count((1,))
124
- labels = ['Pest Swarm Detected', 'No Pest Swarm Detected']
125
- sizes = [count_0, count_1]
126
- plt.figure(figsize=(6, 6))
127
- plt.pie(sizes, labels=labels, autopct='%1.1f%%')
128
- plt.title('IR Value Distribution')
129
- plt.axis('equal')
130
- ir_buffer = io.BytesIO()
131
- plt.savefig(ir_buffer, format='png')
132
- ir_buffer.seek(0)
133
- ir_plot_data = base64.b64encode(ir_buffer.getvalue()).decode('utf-8')
134
- plt.close()
135
-
136
- # Fetch data for average closeness of pests plot
137
- cursor.execute("SELECT timestamp, distance FROM sensor_data WHERE distance < 500")
138
- data = cursor.fetchall()
139
- timestamps = [row[0] for row in data]
140
- distances = [row[1] for row in data]
141
- average_distance = sum(distances) / len(distances)
142
- plt.figure(figsize=(10, 6))
143
- plt.plot(timestamps, distances, marker='o', linestyle='-')
144
- plt.xlabel('Timestamp')
145
- plt.ylabel('Distance (cm)')
146
- plt.title('Average Closeness of Pests over Time')
147
- plt.xticks(rotation=45)
148
- plt.grid(True)
149
- distance_buffer = io.BytesIO()
150
- plt.savefig(distance_buffer, format='png')
151
- distance_buffer.seek(0)
152
- distance_plot_data = base64.b64encode(distance_buffer.getvalue()).decode('utf-8')
153
- plt.close()
154
-
155
- return render_template('activity-log.html', ir_plot_data=ir_plot_data, distance_plot_data=distance_plot_data)
156
-
157
- @app.route('/sensor-data', methods=['POST'])
158
- def update_sensor_data():
159
- data = request.get_json()
160
- temperature = data.get('temperature')
161
- humidity = data.get('humidity')
162
- ir_value = data.get('irValue')
163
- distance = data.get('distance')
164
- sql = "INSERT INTO sensor_data (temperature, humidity, ir_value, distance) VALUES (%s, %s, %s, %s)"
165
- val = (temperature, humidity, ir_value, distance)
166
- cursor.execute(sql, val)
167
- db.commit()
168
- return jsonify(success=True)
169
-
170
- if __name__ == '__main__':
171
- app.run(debug=True, host='0.0.0.0', port=5000)
 
 
1
+ from flask import Flask, render_template, request, jsonify, send_from_directory
2
+ import pymysql
3
+ import matplotlib.pyplot as plt
4
+ import io
5
+ import base64
6
+ from gtts import gTTS
7
+ import os
8
+ import random
9
+ from ai71 import AI71
10
+
11
+ app = Flask(__name__)
12
+ db = pymysql.connect(host="localhost", user="root", password="1234", database="tea_farm")
13
+ cursor = db.cursor()
14
+
15
+ UPLOAD_FOLDER = 'uploads'
16
+ OUTPUT_DIRECTORY = 'audio'
17
+ if not os.path.exists(UPLOAD_FOLDER):
18
+ os.makedirs(UPLOAD_FOLDER)
19
+ if not os.path.exists(OUTPUT_DIRECTORY):
20
+ os.makedirs(OUTPUT_DIRECTORY)
21
+
22
+ # Text for pest and disease
23
+ pest_text = '''tea mosquito bugs,to get rid,spray with neem oil and garlic,use sticky traps, and trap crops like castor . Expect results in 2 to 4 weeks.চা মশার বাগ, পরিত্রাণ পেতে, নিম তেল এবং রসুন দিয়ে স্প্রে করুন, আঠালো ফাঁদ ব্যবহার করুন এবং ক্যাস্টরের মতো ফসল ফাঁদ করুন। 2-4 সপ্তাহের মধ্যে ফলাফল আশা করুন।'''
24
+ disease_text = '''Detected Disease: Brown Blight is a fungal disease To treat it spray a mix of baking soda, water and apply organic mulch to keep soil healthy.ব্রাউন ব্লাইট একটি ছত্রাকজনিত রোগের চিকিৎসার জন্য বেকিং সোডা, পানির মিশ্রণ স্প্রে করুন এবং মাটি সুস্থ রাখতে জৈব মালচ প্রয়োগ করুন'''
25
+
26
+ @app.route('/home')
27
+ def home():
28
+ return render_template('index.html')
29
+
30
+ @app.route('/notification')
31
+ def notification():
32
+ return render_template('notification.html')
33
+
34
+ @app.route('/', methods=['GET', 'POST'])
35
+ def upload_file():
36
+ pest_filename = None
37
+ disease_filename = None
38
+ pest_audio_file = "pest.mp3"
39
+ disease_audio_file = "disease.mp3"
40
+
41
+ if request.method == 'POST':
42
+ if 'pest_file' in request.files:
43
+ pest_file = request.files['pest_file']
44
+ if pest_file.filename != '':
45
+ pest_filename = pest_file.filename
46
+ pest_file.save(os.path.join(UPLOAD_FOLDER, pest_filename))
47
+ tts_pest = gTTS(text=pest_text, lang='en', slow=False)
48
+ tts_pest.save(os.path.join(OUTPUT_DIRECTORY, pest_audio_file))
49
+
50
+ if 'disease_file' in request.files:
51
+ disease_file = request.files['disease_file']
52
+ if disease_file.filename != '':
53
+ disease_filename = disease_file.filename
54
+ disease_file.save(os.path.join(UPLOAD_FOLDER, disease_filename))
55
+ tts_disease = gTTS(text=disease_text, lang='en', slow=False)
56
+ tts_disease.save(os.path.join(OUTPUT_DIRECTORY, disease_audio_file))
57
+
58
+ return render_template('index.html', pest_filename=pest_filename, disease_filename=disease_filename,
59
+ pest_audio_file=pest_audio_file, disease_audio_file=disease_audio_file)
60
+
61
+ AI71_API_KEY = "api71-api-20725a9d-46d6-4baf-9e26-abfca35ab242"
62
+ @app.route('/chat', methods=['POST'])
63
+ def chat():
64
+ message = request.json['message']
65
+ ai71 = AI71(AI71_API_KEY)
66
+ response = ""
67
+ for chunk in ai71.chat.completions.create(
68
+ model="tiiuae/falcon-180b-chat",
69
+ messages=[
70
+ {"role": "system", "content": "You are a helpful assistant for a farming system.Guide the user properly to increase the yield."},
71
+ {"role": "user", "content": message},
72
+ ],
73
+ stream=True,
74
+ ):
75
+ if chunk.choices[0].delta.content:
76
+ response += chunk.choices[0].delta.content
77
+ return jsonify({'response': response.replace('\n','<br>')})
78
+ @app.route('/uploads/<filename>')
79
+ def uploaded_file(filename):
80
+ return send_from_directory(UPLOAD_FOLDER, filename)
81
+
82
+ @app.route('/audio/<filename>')
83
+ def audio_file(filename):
84
+ return send_from_directory(OUTPUT_DIRECTORY, filename)
85
+
86
+ @app.route('/sensors')
87
+ def sensors():
88
+ # Fetch a random row from the database
89
+ cursor.execute("SELECT * FROM sensor_data ORDER BY RAND() LIMIT 1")
90
+ sensor_data = cursor.fetchone()
91
+
92
+ if sensor_data:
93
+ sensor_dict = {
94
+ 'id': sensor_data[0],
95
+ 'temperature': sensor_data[1],
96
+ 'humidity': sensor_data[2],
97
+ 'irValue': sensor_data[3],
98
+ 'distance': sensor_data[4],
99
+ 'timestamp': sensor_data[5]
100
+ }
101
+ else:
102
+ sensor_dict = {
103
+ 'temperature': None,
104
+ 'humidity': None,
105
+ 'irValue': None,
106
+ 'distance': None,
107
+ 'timestamp': None
108
+ }
109
+
110
+ return render_template('sensors.html', sensor_data=sensor_dict)
111
+
112
+ @app.route('/zones')
113
+ def zones():
114
+ return render_template('zones.html')
115
+
116
+ @app.route('/activity-log')
117
+ def activity_log():
118
+
119
+ # Fetch data for IR value distribution plot
120
+ cursor.execute("SELECT ir_value FROM sensor_data")
121
+ ir_values = cursor.fetchall()
122
+ count_0 = ir_values.count((0,))
123
+ count_1 = ir_values.count((1,))
124
+ labels = ['Pest Swarm Detected', 'No Pest Swarm Detected']
125
+ sizes = [count_0, count_1]
126
+ plt.figure(figsize=(6, 6))
127
+ plt.pie(sizes, labels=labels, autopct='%1.1f%%')
128
+ plt.title('IR Value Distribution')
129
+ plt.axis('equal')
130
+ ir_buffer = io.BytesIO()
131
+ plt.savefig(ir_buffer, format='png')
132
+ ir_buffer.seek(0)
133
+ ir_plot_data = base64.b64encode(ir_buffer.getvalue()).decode('utf-8')
134
+ plt.close()
135
+
136
+ # Fetch data for average closeness of pests plot
137
+ cursor.execute("SELECT timestamp, distance FROM sensor_data WHERE distance < 500")
138
+ data = cursor.fetchall()
139
+ timestamps = [row[0] for row in data]
140
+ distances = [row[1] for row in data]
141
+ average_distance = sum(distances) / len(distances)
142
+ plt.figure(figsize=(10, 6))
143
+ plt.plot(timestamps, distances, marker='o', linestyle='-')
144
+ plt.xlabel('Timestamp')
145
+ plt.ylabel('Distance (cm)')
146
+ plt.title('Average Closeness of Pests over Time')
147
+ plt.xticks(rotation=45)
148
+ plt.grid(True)
149
+ distance_buffer = io.BytesIO()
150
+ plt.savefig(distance_buffer, format='png')
151
+ distance_buffer.seek(0)
152
+ distance_plot_data = base64.b64encode(distance_buffer.getvalue()).decode('utf-8')
153
+ plt.close()
154
+
155
+ return render_template('activity-log.html', ir_plot_data=ir_plot_data, distance_plot_data=distance_plot_data)
156
+
157
+ @app.route('/sensor-data', methods=['POST'])
158
+ def update_sensor_data():
159
+ data = request.get_json()
160
+ temperature = data.get('temperature')
161
+ humidity = data.get('humidity')
162
+ ir_value = data.get('irValue')
163
+ distance = data.get('distance')
164
+ sql = "INSERT INTO sensor_data (temperature, humidity, ir_value, distance) VALUES (%s, %s, %s, %s)"
165
+ val = (temperature, humidity, ir_value, distance)
166
+ cursor.execute(sql, val)
167
+ db.commit()
168
+ return jsonify(success=True)
169
+
170
+ if __name__ == "__main__":
171
+
172
+ app.run(host='0.0.0.0', port=7860,debug=1==1)