Update app.py

app.py

@@ -794,21 +794,34 @@ class MyHandler(SimpleHTTPRequestHandler):
 
     def do_POST(self):
         if self.path == '/generate_goals':
-            content_length = int(self.headers['Content-Length'])
-            post_data = self.rfile.read(content_length)
-            data = json.loads(post_data.decode('utf-8'))
-            input_var = data['input_var']
-            goals = generate_goals(input_var)
-            
-            self.send_response(200)
-            self.send_header('Content-type', 'application/json')
-            self.end_headers()
-            self.wfile.write(json.dumps({'goals': goals}).encode())
+            try:
+                content_length = int(self.headers['Content-Length'])
+                post_data = self.rfile.read(content_length)
+                data = json.loads(post_data.decode('utf-8'))
+                input_var = data['input_var']
+                goals = generate_goals(input_var)
+                
+                response = json.dumps({'goals': goals}).encode()
+                self.send_response(200)
+                self.send_header('Content-type', 'application/json')
+                self.send_header('Content-Length', str(len(response)))
+                self.end_headers()
+                self.wfile.write(response)
+            except Exception as e:
+                logging.error(f"Error handling POST request: {str(e)}")
+                self.send_error(500, f"Internal server error: {str(e)}")
         else:
             self.send_error(404)
 
+def run_server(port):
+    try:
+        server = ThreadingHTTPServer(('0.0.0.0', port), MyHandler)
+        print(f"Server running on port {port}")
+        server.serve_forever()
+    except Exception as e:
+        logging.error(f"Error running server: {str(e)}")
+        traceback.print_exc()
+
 if __name__ == '__main__':
     port = int(os.environ.get("PORT", 7860))
-    server = HTTPServer(('0.0.0.0', port), MyHandler)
-    print(f"Server running on port {port}")
-    server.serve_forever()
+    run_server(port)