Deepakraj2006 commited on
Commit
03a8877
·
verified ·
1 Parent(s): 5c1da39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -14
app.py CHANGED
@@ -12,12 +12,11 @@ app.logger.setLevel(logging.ERROR)
12
  # Ensure worker is initialized
13
  worker.init_llm()
14
 
15
- # ✅ Create an uploads directory if it doesn't exist (with proper permissions)
16
- UPLOAD_FOLDER = "./uploads"
17
 
18
  try:
19
  os.makedirs(UPLOAD_FOLDER, exist_ok=True)
20
- os.chmod(UPLOAD_FOLDER, 0o777) # Ensure full read/write permissions
21
  print(f"Uploads directory created or exists: {UPLOAD_FOLDER}")
22
  except Exception as e:
23
  print(f"Error creating uploads directory: {str(e)}")
@@ -25,36 +24,34 @@ except Exception as e:
25
  # Define the route for the index page
26
  @app.route('/', methods=['GET'])
27
  def index():
28
- return render_template('index.html') # Render the index.html template
29
 
30
  # Define the route for processing messages
31
  @app.route('/process-message', methods=['POST'])
32
  def process_message_route():
33
- user_message = request.json.get('userMessage', '') # Extract the user's message from the request
34
 
35
  if not user_message:
36
  return jsonify({"botResponse": "Please enter a valid message."}), 400
37
 
38
- bot_response = worker.process_prompt(user_message) # Process the user's message using the worker module
39
 
40
- # Return the bot's response as JSON
41
  return jsonify({"botResponse": bot_response}), 200
42
 
43
  # Define the route for processing documents
44
  @app.route('/process-document', methods=['POST'])
45
  def process_document_route():
46
- # Check if a file was uploaded
47
  if 'file' not in request.files:
48
  return jsonify({
49
  "botResponse": "It seems like the file was not uploaded correctly. Please try again."
50
  }), 400
51
 
52
- file = request.files['file'] # Extract the uploaded file from the request
53
- file_path = os.path.join(UPLOAD_FOLDER, file.filename) # Save file in the uploads directory
54
 
55
  try:
56
- file.save(file_path) # Save the file
57
- worker.process_document(file_path) # Process the document using the worker module
58
 
59
  return jsonify({
60
  "botResponse": "Thank you for providing your PDF document. I have analyzed it, so now you can ask me any "
@@ -63,6 +60,5 @@ def process_document_route():
63
  except Exception as e:
64
  return jsonify({"botResponse": f"Error saving the file: {str(e)}"}), 500
65
 
66
- # Run the Flask app
67
  if __name__ == "__main__":
68
- app.run(debug=True, port=8000, host='0.0.0.0')
 
12
  # Ensure worker is initialized
13
  worker.init_llm()
14
 
15
+ # ✅ Use `/tmp/uploads/` instead of `./uploads` (since `/tmp/` is writable)
16
+ UPLOAD_FOLDER = "/tmp/uploads"
17
 
18
  try:
19
  os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
20
  print(f"Uploads directory created or exists: {UPLOAD_FOLDER}")
21
  except Exception as e:
22
  print(f"Error creating uploads directory: {str(e)}")
 
24
  # Define the route for the index page
25
  @app.route('/', methods=['GET'])
26
  def index():
27
+ return render_template('index.html')
28
 
29
  # Define the route for processing messages
30
  @app.route('/process-message', methods=['POST'])
31
  def process_message_route():
32
+ user_message = request.json.get('userMessage', '')
33
 
34
  if not user_message:
35
  return jsonify({"botResponse": "Please enter a valid message."}), 400
36
 
37
+ bot_response = worker.process_prompt(user_message)
38
 
 
39
  return jsonify({"botResponse": bot_response}), 200
40
 
41
  # Define the route for processing documents
42
  @app.route('/process-document', methods=['POST'])
43
  def process_document_route():
 
44
  if 'file' not in request.files:
45
  return jsonify({
46
  "botResponse": "It seems like the file was not uploaded correctly. Please try again."
47
  }), 400
48
 
49
+ file = request.files['file']
50
+ file_path = os.path.join(UPLOAD_FOLDER, file.filename)
51
 
52
  try:
53
+ file.save(file_path)
54
+ worker.process_document(file_path)
55
 
56
  return jsonify({
57
  "botResponse": "Thank you for providing your PDF document. I have analyzed it, so now you can ask me any "
 
60
  except Exception as e:
61
  return jsonify({"botResponse": f"Error saving the file: {str(e)}"}), 500
62
 
 
63
  if __name__ == "__main__":
64
+ app.run(debug=False, port=8000, host='0.0.0.0')